Skip to content

[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1081

[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob

[Feature] Force log_prob to return a tensordict when kwargs are passed to ProbabilisticTensorDictSequential.log_prob #1081

Annotations

1 error and 1 warning

pytorch/tensordict  /  wheel-py3_9-cuda12_4

succeeded Dec 18, 2024 in 10m 41s