diff --git a/main.py b/main.py index f46e0ba..2609971 100644 --- a/main.py +++ b/main.py @@ -114,6 +114,7 @@ for i_episode in itertools.count(1): if i_episode % 10 == 0 and args.eval == True: state = env.reset() episode_reward = 0 + done = False while not done: action = agent.select_action(state, eval=True)