From e06d7810b71c57d82e60ec4747177526df517cb4 Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Sun, 5 Nov 2017 13:02:52 +0800 Subject: [PATCH] Update continuous_action_network.py --- network/continuous_action_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/network/continuous_action_network.py b/network/continuous_action_network.py index b9fcb42..1bb3efc 100644 --- a/network/continuous_action_network.py +++ b/network/continuous_action_network.py @@ -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):