From 9ccf247dfba8306c39fdba2217e7effe6606a807 Mon Sep 17 00:00:00 2001 From: Mike Clark Date: Sun, 5 Nov 2017 12:57:23 +0800 Subject: [PATCH] entropy equation? --- network/continuous_action_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/network/continuous_action_network.py b/network/continuous_action_network.py index d900094..b9fcb42 100644 --- a/network/continuous_action_network.py +++ b/network/continuous_action_network.py @@ -158,7 +158,7 @@ class GaussianActorNet(nn.Module, BasicNet): return log_density.sum(1) def entropy(self, std): - return 0.5 * (1 + (2 * std.pow(2) * np.pi + 1e-5).log()).sum(1).mean() + return 0.5 * (2 * var * np.pi * np.e).log().sum(1).mean() class GaussianCriticNet(nn.Module, BasicNet): def __init__(self, state_dim, gpu=False):