From e9c0521ff5f3413e4205f2efeb2a98a7c689b834 Mon Sep 17 00:00:00 2001 From: Mark Date: Fri, 24 Feb 2023 14:22:08 -0800 Subject: [PATCH] Changed acrobot to cartpole --- polecart/main/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/polecart/main/main.py b/polecart/main/main.py index cd7ca59..8debc5a 100755 --- a/polecart/main/main.py +++ b/polecart/main/main.py @@ -74,7 +74,7 @@ if __name__ == '__main__': # CartPole-v1 # Acrobot-v1 agent = Agent( - env_name = "Acrobot-v1", + env_name = "CartPole-v1", network = DQN, BATCH_SIZE = 128, TAU = 0.005,