Merge pull request #7 from jendelel/init_eval_state_bug

Fixed bug with double tensoring the initial state.
2026-06-27 19:00:26 +08:00 · 2019-03-25 23:33:36 +05:30
parent 3072d6b727 8ab8cd61f9
commit 9e1eda980f
1 changed files with 1 additions and 1 deletions
@@ -109,7 +109,7 @@ for i_episode in itertools.count():
                                                                                np.round(np.mean(rewards[-100:]),2)))

    if i_episode % 10 == 0 and args.eval == True:
-        state = torch.Tensor([env.reset()])
+        state = env.reset()
        episode_reward = 0
        while True:
            action = agent.select_action(state, eval=True)