[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #2444
Job | Run time |
---|---|
13m 6s | |
12m 58s | |
13m 13s | |
13m 48s | |
18m 59s | |
13m 40s | |
18m 39s | |
13m 7s | |
1h 57m 30s |
Job | Run time |
---|---|
13m 6s | |
12m 58s | |
13m 13s | |
13m 48s | |
18m 59s | |
13m 40s | |
18m 39s | |
13m 7s | |
1h 57m 30s |