diff --git a/network/continuous_action_network.py b/network/continuous_action_network.py index e322b5f..a77281c 100644 --- a/network/continuous_action_network.py +++ b/network/continuous_action_network.py @@ -145,7 +145,7 @@ class GaussianActorNet(nn.Module, BasicNet): log_std = self.action_log_std.expand_as(mean) std = log_std.exp() else: - std = F.softplus(self.action_std(phi) + 1e-5) + std = F.softplus(self.action_std(phi)) + 1e-5 log_std = std.log() return mean, std, log_std