From e9cc3fd7e8b67a58fb27ddb59cb57fd8d70af7d9 Mon Sep 17 00:00:00 2001 From: pranz24 Date: Mon, 20 May 2019 13:29:23 +0530 Subject: [PATCH] Add Normalized Actions --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 7d19e5f..21f8b17 100644 --- a/main.py +++ b/main.py @@ -123,10 +123,10 @@ for i_episode in itertools.count(1): avg_reward /= episodes - writer.add_scalar('reward/test', episode_reward, i_episode) + writer.add_scalar('reward/test', avg_reward, i_episode) print("----------------------------------------") - print("Test Episodes: {}, Avg. Reward: {}".format(i_episode, round(episode_reward, 2))) + print("Test Episodes: {}, Avg. Reward: {}".format(episodes, round(avg_reward, 2))) print("----------------------------------------") env.close()