Update model.py

This commit is contained in:
Pranjal Tandon
2019-04-05 01:05:36 +05:30
committed by GitHub
parent 7e1496f87f
commit ee617f3fed
+1 -1
View File
@@ -118,5 +118,5 @@ class DeterministicPolicy(nn.Module):
noise = self.noise.normal_(0., std=0.1)
noise = noise.clamp(-0.25, 0.25)
action = mean + noise
return action, torch.tensor(0.), torch.tensor(0.), mean, torch.tensor(0.)
return action, mean, torch.tensor(0.)