From 076f46707db9d181e4cea19f563d7baad2bd34e8 Mon Sep 17 00:00:00 2001 From: pranz24 Date: Mon, 20 May 2019 13:26:49 +0530 Subject: [PATCH] Add Normalized Actions --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 992c508..8067a6b 100644 --- a/main.py +++ b/main.py @@ -132,7 +132,7 @@ for i_episode in itertools.count(1): avg_reward /= episodes - writer.add_scalar('reward/test', episode_reward, i_episode) + writer.add_scalar('avg_reward/test', avg_reward, i_episode) print("----------------------------------------") print("Test Episodes: {}, Avg. Reward: {}".format(episodes, round(avg_reward, 2)))