hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix: Add MujocoEnvObsRmsPersistence only if obs_norm is enabled

Browse Source
This commit is contained in:
Dominik Jain 2023-10-24 13:52:30 +02:00
parent 58466ebf5d
commit dd4a0eb430
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/mujoco/mujoco_env.py
View File

@ -83,5 +83,6 @@ class MujocoEnvFactory(EnvFactory):
obs_norm=self.obs_norm, obs_norm=self.obs_norm,
) )
envs = ContinuousEnvironments(env=env, train_envs=train_envs, test_envs=test_envs) envs = ContinuousEnvironments(env=env, train_envs=train_envs, test_envs=test_envs)
if self.obs_norm:
envs.set_persistence(MujocoEnvObsRmsPersistence()) envs.set_persistence(MujocoEnvObsRmsPersistence())
return envs return envs