Skip to content

Commit f3f58bb

Browse files
authored
Update gym.wrapper.Monitor()
1 parent 1074a3e commit f3f58bb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

DQN/Deep Q Learning Solution.ipynb

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,8 @@
403403
"\n",
404404
" # Record videos\n",
405405
" # Add env Monitor wrapper\n",
406-
" env = Monitor(env, directory=monitor_path, video_callable=lambda x: True, resume=True)\n",
406+
" env = Monitor(env, directory=monitor_path, video_callable=lambda count: count % record_video_every == 0, resume=True)\n",
407+
"\n",
407408
" for i_episode in range(num_episodes):\n",
408409
"\n",
409410
" # Save the current checkpoint\n",

0 commit comments

Comments
 (0)