From 105e9528feba023cad70327f3d2bdaf93cba8b27 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alperen=20=C3=9CNL=C3=9C?= <97191996+alperenunlu@users.noreply.github.com> Date: Tue, 7 Nov 2023 19:26:52 +0300 Subject: [PATCH] Add logging to last episode (#2656) --- intermediate_source/mario_rl_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intermediate_source/mario_rl_tutorial.py b/intermediate_source/mario_rl_tutorial.py index 20f1b61877..ffb4a54ac0 100755 --- 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)