From 3d1ce87d70f99625cebafb67322bd40cc43ee0bb Mon Sep 17 00:00:00 2001 From: xiaoyisd Date: Fri, 12 Mar 2021 16:37:27 +0800 Subject: [PATCH] cartpolev0-cartpolev1 --- model_zoo/official/rl/dqn/eval.py | 2 +- model_zoo/official/rl/dqn/train.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/model_zoo/official/rl/dqn/eval.py b/model_zoo/official/rl/dqn/eval.py index 59d51e9b716..7f61abf4772 100644 --- a/model_zoo/official/rl/dqn/eval.py +++ b/model_zoo/official/rl/dqn/eval.py @@ -33,7 +33,7 @@ set_seed(1) if __name__ == "__main__": context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target) - env = gym.make('CartPole-v0') + env = gym.make('CartPole-v1') cfg.state_space_dim = env.observation_space.shape[0] cfg.action_space_dim = env.action_space.n agent = Agent(**cfg) diff --git a/model_zoo/official/rl/dqn/train.py b/model_zoo/official/rl/dqn/train.py index 919c7193fd3..42f56df737c 100644 --- a/model_zoo/official/rl/dqn/train.py +++ b/model_zoo/official/rl/dqn/train.py @@ -34,13 +34,13 @@ set_seed(1) if __name__ == "__main__": context.set_context(mode=context.GRAPH_MODE, device_target=args.device_target) - env = gym.make('CartPole-v0') + env = gym.make('CartPole-v1') cfg.state_space_dim = env.observation_space.shape[0] cfg.action_space_dim = env.action_space.n agent = Agent(**cfg) agent.load_dict() - for episode in range(150): + for episode in range(300): s0 = env.reset() total_reward = 1 while True: