Mark
/
celeste-ai
Archived
1
0
Fork 0

Changed acrobot to cartpole

master
Mark 2023-02-24 14:22:08 -08:00
parent b3a252d970
commit e9c0521ff5
Signed by: Mark
GPG Key ID: AD62BB059C2AAEE4
1 changed files with 1 additions and 1 deletions

View File

@ -74,7 +74,7 @@ if __name__ == '__main__':
# CartPole-v1 # CartPole-v1
# Acrobot-v1 # Acrobot-v1
agent = Agent( agent = Agent(
env_name = "Acrobot-v1", env_name = "CartPole-v1",
network = DQN, network = DQN,
BATCH_SIZE = 128, BATCH_SIZE = 128,
TAU = 0.005, TAU = 0.005,