diff --git a/cleanrl/ppo_atari.py b/cleanrl/ppo_atari.py index 2277faf9..87c15a23 100644 --- a/cleanrl/ppo_atari.py +++ b/cleanrl/ppo_atari.py @@ -94,9 +94,6 @@ def thunk(): else: env = gym.make(env_id) env = gym.wrappers.RecordEpisodeStatistics(env) - if capture_video: - if idx == 0: - env = gym.wrappers.RecordVideo(env, f"videos/{run_name}") env = NoopResetEnv(env, noop_max=30) env = MaxAndSkipEnv(env, skip=4) env = EpisodicLifeEnv(env) diff --git a/cleanrl/ppo_atari_lstm.py b/cleanrl/ppo_atari_lstm.py index 0848074c..154146da 100644 --- a/cleanrl/ppo_atari_lstm.py +++ b/cleanrl/ppo_atari_lstm.py @@ -94,9 +94,6 @@ def thunk(): else: env = gym.make(env_id) env = gym.wrappers.RecordEpisodeStatistics(env) - if capture_video: - if idx == 0: - env = gym.wrappers.RecordVideo(env, f"videos/{run_name}") env = NoopResetEnv(env, noop_max=30) env = MaxAndSkipEnv(env, skip=4) env = EpisodicLifeEnv(env) diff --git a/cleanrl/ppo_atari_multigpu.py b/cleanrl/ppo_atari_multigpu.py index c8313e2e..7a37b4ee 100644 --- a/cleanrl/ppo_atari_multigpu.py +++ b/cleanrl/ppo_atari_multigpu.py @@ -110,9 +110,6 @@ def thunk(): else: env = gym.make(env_id) env = gym.wrappers.RecordEpisodeStatistics(env) - if capture_video: - if idx == 0: - env = gym.wrappers.RecordVideo(env, f"videos/{run_name}") env = NoopResetEnv(env, noop_max=30) env = MaxAndSkipEnv(env, skip=4) env = EpisodicLifeEnv(env)