From 8adb8f9931b1b8ba668f73d33e2d640439bb59aa Mon Sep 17 00:00:00 2001
From: Antonin RAFFIN <antonin.raffin@ensta.org>
Date: Sun, 22 Sep 2019 12:56:27 +0200
Subject: [PATCH] Change default dist to gaussian

---
 torchy_baselines/ppo/policies.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/torchy_baselines/ppo/policies.py b/torchy_baselines/ppo/policies.py
index 3ed9a28..bc0d48b 100644
--- a/torchy_baselines/ppo/policies.py
+++ b/torchy_baselines/ppo/policies.py
@@ -29,8 +29,8 @@ class PPOPolicy(BasePolicy):
         self.shared_net = None
         self.pi_net, self.vf_net = None, None
         # Action distribution
-        # self.action_dist = DiagGaussianDistribution(self.action_dim)
-        self.action_dist = SquashedDiagGaussianDistribution(self.action_dim)
+        self.action_dist = DiagGaussianDistribution(self.action_dim)
+        # self.action_dist = SquashedDiagGaussianDistribution(self.action_dim)
         self._build(learning_rate)
 
     @staticmethod