saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-07 17:15:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Change default dist to gaussian

Browse source
This commit is contained in:
Antonin RAFFIN 2019-09-22 12:56:27 +02:00
parent ddaafcbc36
commit 8adb8f9931
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torchy_baselines/ppo/policies.py
View file

@ -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