[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1081
Job | Run time |
---|---|
4s | |
7m 53s | |
11m 2s | |
10m 41s | |
10m 33s | |
13s | |
13s | |
13s | |
12s | |
41m 4s |
Job | Run time |
---|---|
4s | |
7m 53s | |
11m 2s | |
10m 41s | |
10m 33s | |
13s | |
13s | |
13s | |
12s | |
41m 4s |