flip sign error

berkeleydeeprlcourse · Oct 14, 2020 · d28416b · d28416b
1 parent 291daf6
commit d28416b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/hw3/cs285/infrastructure/rl_trainer.py b/hw3/cs285/infrastructure/rl_trainer.py
@@ -55,12 +55,13 @@ def __init__(self, params):
         if 'env_wrappers' in self.params:
             # These operations are currently only for Atari envs
             self.env = wrappers.Monitor(self.env, os.path.join(self.params['logdir'], "gym"), force=True)
-            self.env.enabled = (self.params['video_log_freq'] <= 0)
+            self.env.enabled = (self.params['video_log_freq'] > 0)
             self.env = params['env_wrappers'](self.env)
             self.mean_episode_reward = -float('nan')
             self.best_mean_episode_reward = -float('inf')
         if 'non_atari_colab_env' in self.params and self.params['video_log_freq'] > 0:
             self.env = wrappers.Monitor(self.env, os.path.join(self.params['logdir'], "gym"), force=True)
+            self.env.enabled = (self.params['video_log_freq'] > 0)
             self.mean_episode_reward = -float('nan')
             self.best_mean_episode_reward = -float('inf')