Merge branch 'main' into jingxu10/2629

pytorch · Nov 7, 2023 · a5eee63 · a5eee63
2 parents 36a8907 + 105e952
commit a5eee63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intermediate_source/mario_rl_tutorial.py b/intermediate_source/mario_rl_tutorial.py
@@ -778,7 +778,7 @@ def record(self, episode, epsilon, step):
 
     logger.log_episode()
 
-    if e % 20 == 0:
+    if (e % 20 == 0) or (e == episodes - 1):
         logger.record(episode=e, epsilon=mario.exploration_rate, step=mario.curr_step)