[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1079
Job | Run time |
---|---|
4s | |
2m 49s | |
6m 1s | |
6m 38s | |
5m 59s | |
13m 33s | |
15m 47s | |
9s | |
13s | |
12s | |
9s | |
12s | |
11s | |
51m 57s |
Job | Run time |
---|---|
4s | |
2m 49s | |
6m 1s | |
6m 38s | |
5m 59s | |
13m 33s | |
15m 47s | |
9s | |
13s | |
12s | |
9s | |
12s | |
11s | |
51m 57s |