jostyposty
commited on
Commit
•
f88882e
1
Parent(s):
3261e0d
fix: rename video to replay.mp4
Browse files- main.py +1 -1
- replay.mp4 +0 -0
- video.mp4 +0 -0
main.py
CHANGED
@@ -43,7 +43,7 @@ generate_results_json("results.json", mean_reward, std_reward, n_eval_episodes,
|
|
43 |
|
44 |
# 4. Generate video
|
45 |
model_for_video = load_ppo_model_for_video(main_model_fp, env_id)
|
46 |
-
generate_video(model_for_video, "
|
47 |
|
48 |
|
49 |
# 5. Generate model card
|
|
|
43 |
|
44 |
# 4. Generate video
|
45 |
model_for_video = load_ppo_model_for_video(main_model_fp, env_id)
|
46 |
+
generate_video(model_for_video, "replay.mp4", video_length_in_episodes=5)
|
47 |
|
48 |
|
49 |
# 5. Generate model card
|
replay.mp4
ADDED
Binary file (131 kB). View file
|
|
video.mp4
DELETED
Binary file (143 kB)
|
|