[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1080
Job | Run time |
---|---|
6s | |
3m 30s | |
5m 37s | |
6m 3s | |
5m 39s | |
13m 40s | |
16m 8s | |
13s | |
14s | |
14s | |
12s | |
15s | |
14s | |
52m 5s |
Job | Run time |
---|---|
6s | |
3m 30s | |
5m 37s | |
6m 3s | |
5m 39s | |
13m 40s | |
16m 8s | |
13s | |
14s | |
14s | |
12s | |
15s | |
14s | |
52m 5s |