From 8ab8cd61f93007e46ab4e3bdb8ef6179eca31b1c Mon Sep 17 00:00:00 2001 From: Lukas Jendele Date: Mon, 25 Mar 2019 16:39:13 +0100 Subject: [PATCH] Fixed bug with double tensoring the initial state. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 9bf85ce..97a6b1c 100644 --- a/main.py +++ b/main.py @@ -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)