From 73064f31ea4f1b2b46de59c5f3d2060d4f15748e Mon Sep 17 00:00:00 2001 From: Pranjal Tandon Date: Mon, 3 Feb 2020 13:46:39 +0530 Subject: [PATCH] Update main.py --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 7fca690..80ba480 100644 --- a/main.py +++ b/main.py @@ -111,7 +111,7 @@ for i_episode in itertools.count(1): writer.add_scalar('reward/train', episode_reward, i_episode) print("Episode: {}, total numsteps: {}, episode steps: {}, reward: {}".format(i_episode, total_numsteps, episode_steps, round(episode_reward, 2))) - if i_episode % 10 == 0 and args.eval == True: + if i_episode % 10 == 0 and args.eval is True: avg_reward = 0. episodes = 10 for _ in range(episodes):