Changed acrobot to cartpole
parent
b3a252d970
commit
e9c0521ff5
|
@ -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,
|
||||
|
|
Reference in New Issue