From e1e6446728d0a9b302e3c0e1ca8b6421c9ffc5d2 Mon Sep 17 00:00:00 2001 From: Shangtong Zhang Date: Thu, 9 Nov 2017 08:50:57 -0700 Subject: [PATCH] Minor update --- main.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/main.py b/main.py index ddae220..be7c505 100644 --- a/main.py +++ b/main.py @@ -199,11 +199,12 @@ def a3c_continuous(): agent = AsyncAgent(config) agent.run() -def dppo_continuous(): +def p3o_continuous(): config = Config() config.task_fn = lambda: Pendulum() # config.task_fn = lambda: BipedalWalkerHardcore() # config.task_fn = lambda: Roboschool('RoboschoolInvertedPendulum-v1') + # config.task_fn = lambda: Roboschool('RoboschoolAnt-v1') task = config.task_fn() config.actor_network_fn = lambda: GaussianActorNet(task.state_dim, task.action_dim, gpu=False, unit_std=True) @@ -263,11 +264,11 @@ if __name__ == '__main__': # gym.logger.setLevel(logging.DEBUG) gym.logger.setLevel(logging.INFO) - dqn_cart_pole() + # dqn_cart_pole() # async_cart_pole() # a3c_cart_pole() # a3c_continuous() - # dppo_continuous() + p3o_continuous() # ddpg_continuous() # dqn_fruit()