ledmands commited on
Commit
956eab4
1 Parent(s): 7a9dcf1

Removed testing videos

Browse files
agents/videos/rl-video-step-0-to-step-10000.meta.json DELETED
@@ -1 +0,0 @@
1
- {"step_id": 0, "content_type": "video/mp4"}
 
 
agents/videos/rl-video-step-0-to-step-10000.mp4 DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1ca60ce100e54e68edd9e76c84bb7f1bfd6730b2ddea8090b7ef0c1ba675e592
3
- size 149189
 
 
 
 
agents/videos/rl-video-step-0-to-step-100000.meta.json DELETED
@@ -1 +0,0 @@
1
- {"step_id": 0, "content_type": "video/mp4"}
 
 
agents/videos/rl-video-step-0-to-step-100000.mp4 DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:d433123f98e1de6186e713a1890a2ddfc9f2330ace545f5f38b73d3d03f3a0c8
3
- size 174076
 
 
 
 
agents/videos/rl-video-step-0-to-step-200.meta.json DELETED
@@ -1 +0,0 @@
1
- {"step_id": 0, "content_type": "video/mp4"}
 
 
agents/videos/rl-video-step-0-to-step-200.mp4 DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:07ba166ed637c797aef7bc6017b724ae5162ffbb264b57e4f0515140d0e15463
3
- size 33296
 
 
 
 
record_video.py CHANGED
@@ -3,7 +3,7 @@ from stable_baselines3 import DQN
3
  from stable_baselines3.common.monitor import Monitor
4
  from stable_baselines3.common.vec_env import VecVideoRecorder, DummyVecEnv, VecEnv
5
 
6
- model_name = "agents/dqn_v2-7/ALE-Pacman-v5" # path to model, should be an argument
7
  env_id = "ALE/Pacman-v5"
8
  video_folder = "videos/"
9
  video_length = 100000 #steps by hard coding this, I can almost ensure only one episode is recorded...
 
3
  from stable_baselines3.common.monitor import Monitor
4
  from stable_baselines3.common.vec_env import VecVideoRecorder, DummyVecEnv, VecEnv
5
 
6
+ model_name = "agents/dqn_v2-8/best_model" # path to model, should be an argument
7
  env_id = "ALE/Pacman-v5"
8
  video_folder = "videos/"
9
  video_length = 100000 #steps by hard coding this, I can almost ensure only one episode is recorded...