From 3cb499632eda306fb887ba42022de4690564fe6f Mon Sep 17 00:00:00 2001 From: Eugene Vinitsky Date: Fri, 13 Dec 2019 23:42:30 +0100 Subject: [PATCH] (Bug Fix): Remove the extra 0.5 in the Diagonal Gaussian entropy (#6475) --- rllib/models/tf/tf_action_dist.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rllib/models/tf/tf_action_dist.py b/rllib/models/tf/tf_action_dist.py index 9474d2785..27b171c6c 100644 --- a/rllib/models/tf/tf_action_dist.py +++ b/rllib/models/tf/tf_action_dist.py @@ -166,7 +166,7 @@ class DiagGaussian(TFActionDistribution): @override(ActionDistribution) def entropy(self): return tf.reduce_sum( - .5 * self.log_std + .5 * np.log(2.0 * np.pi * np.e), + self.log_std + .5 * np.log(2.0 * np.pi * np.e), reduction_indices=[1]) @override(TFActionDistribution)