[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1080
Job | Run time |
---|---|
6s | |
9m 12s | |
11m 37s | |
12m 16s | |
11m 10s | |
13s | |
9s | |
9s | |
13s | |
45m 5s |
Job | Run time |
---|---|
6s | |
9m 12s | |
11m 37s | |
12m 16s | |
11m 10s | |
13s | |
9s | |
9s | |
13s | |
45m 5s |