[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #2445
Job | Run time |
---|---|
18m 51s | |
19m 22s | |
13m 45s | |
19m 16s | |
19m 0s | |
13m 27s | |
19m 18s | |
26m 23s | |
2h 29m 22s |
Job | Run time |
---|---|
18m 51s | |
19m 22s | |
13m 45s | |
19m 16s | |
19m 0s | |
13m 27s | |
19m 18s | |
26m 23s | |
2h 29m 22s |