Update continuous_action_network.py

2026-09-09 11:13:47 +08:00 · 2017-11-05 13:02:52 +08:00
parent 5ce9264156
commit e06d7810b7
1 changed files with 1 additions and 0 deletions
@@ -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):