wassname/ray
1
0 Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-10-28 13:21:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix TorchDeterministic (#9241)

Browse Source
This commit is contained in:
Raphael Avalos authored and GitHub committed 2020-07-23 10:43:20 -07:00
1 parent 9f969260e8
commit 5303c3abe3
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
rllib/models/torch/torch_action_dist.py
+1 -1
View File
@@ -308,7 +308,7 @@ class TorchDeterministic(TorchDistributionWrapper):
@override(TorchDistributionWrapper)
def sampled_action_logp(self):
return 0.0
return torch.zeros((self.inputs.size()[0], ), dtype=torch.float32)
@override(TorchDistributionWrapper)
def sample(self):