Changed acrobot to cartpole

2023-02-24 14:22:08 -08:00 · 2023-02-24 14:22:08 -08:00 · e9c0521ff5
commit e9c0521ff5
parent b3a252d970
1 changed files with 1 additions and 1 deletions
--- 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,