mirror of
https://github.com/wassname/DeepRL.git
synced 2026-10-28 12:30:38 +08:00
Update continuous_action_network.py
This commit is contained in:
1 parent
5ce9264156
commit
e06d7810b7
1 file changed
+1
@@ -158,6 +158,7 @@ class GaussianActorNet(nn.Module, BasicNet):
|
||||
return log_density.sum(1)
|
||||
|
||||
def entropy(self, std):
|
||||
var = std.pow(2) + 1e-5
|
||||
return 0.5 * (2 * var * np.pi * np.e).log().sum(1).mean()
|
||||
|
||||
class GaussianCriticNet(nn.Module, BasicNet):
|
||||
|
||||
Reference in new issue
Block a user