Changed the video recorder directory

agents/record_video.py → record_video.py

@@ -3,10 +3,10 @@ from stable_baselines3 import DQN
 from stable_baselines3.common.monitor import Monitor
 from stable_baselines3.common.vec_env import VecVideoRecorder, DummyVecEnv, VecEnv
 
-model_name = "dqn_v2-5/ALE-Pacman-v5" # path to model, should be an argument
+model_name = "agents/dqn_v2-7/ALE-Pacman-v5" # path to model, should be an argument
 env_id = "ALE/Pacman-v5"
 video_folder = "videos/"
-video_length = 100000 #steps
+video_length = 100000 #steps by hard coding this, I can almost ensure only one episode is recorded...
 
 vec_env = DummyVecEnv([lambda: gym.make(env_id, render_mode="rgb_array")])
 model = DQN.load(model_name)
@@ -29,7 +29,7 @@ vec_env = VecVideoRecorder(vec_env,
                            video_folder, 
                            record_video_trigger=lambda x: x == 0,
                            video_length=video_length,
-                          #  name_prefix=f"video-{env_id}"
+                           name_prefix=f"video-{model_name}"
                            )
 #  Once I make the environment, now I need to walk through it...???
 #   I want to act according to the policy that has been trained