Skip to content

Commit ee3d8b1

Browse files
authored
Merge branch 'main' into train_eval_fix
2 parents c7b90c3 + 105e952 commit ee3d8b1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/mario_rl_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -778,7 +778,7 @@ def record(self, episode, epsilon, step):
778778

779779
logger.log_episode()
780780

781-
if e % 20 == 0:
781+
if (e % 20 == 0) or (e == episodes - 1):
782782
logger.record(episode=e, epsilon=mario.exploration_rate, step=mario.curr_step)
783783

784784

0 commit comments

Comments
 (0)