wassname/pytorch-soft-actor-critic
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-soft-actor-critic.git synced 2026-09-09 11:32:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21 from Shmuma/patch-1

Browse Source 
Fix error with DeterministicPolicy
This commit is contained in:
Pranjal Tandon
2019-11-27 13:30:57 +05:30
committed by GitHub
parent cc42a1f31c 3664ba4e60
commit d8ba7370e5
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model.py
+2 -1
View File
@@ -148,4 +148,5 @@ class DeterministicPolicy(nn.Module):
def to(self, device):
self.action_scale = self.action_scale.to(device)
self.action_bias = self.action_bias.to(device)
return super(GaussianPolicy, self).to(device)
self.noise = self.noise.to(device)
return super(DeterministicPolicy, self).to(device)