Merge pull request #4 from wassname/patch-1

moving epsilon outside softplus
This commit is contained in:
Shangtong Zhang
2017-10-31 21:01:10 -06:00
committed by GitHub
+1 -1
View File
@@ -145,7 +145,7 @@ class GaussianActorNet(nn.Module, BasicNet):
log_std = self.action_log_std.expand_as(mean)
std = log_std.exp()
else:
std = F.softplus(self.action_std(phi) + 1e-5)
std = F.softplus(self.action_std(phi)) + 1e-5
log_std = std.log()
return mean, std, log_std