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