saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-06-02 23:40:09 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Change default architecture

Browse source
This commit is contained in:
Antonin Raffin 2019-11-22 17:24:14 +01:00
parent 03d2ab10f8
commit f0f2f10d1e
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
torchy_baselines/ppo/policies.py
View file

@ -16,8 +16,12 @@ class PPOPolicy(BasePolicy):
ortho_init=True, use_sde=False, log_std_init=0.0):
super(PPOPolicy, self).__init__(observation_space, action_space, device)
self.obs_dim = self.observation_space.shape[0]
# Default network architecture, from stable-baselines
if net_arch is None:
net_arch = [dict(pi=[64], vf=[64])]
net_arch = [dict(pi=[64, 64], vf=[64, 64])]
self.net_arch = net_arch
self.activation_fn = activation_fn
self.adam_epsilon = adam_epsilon