xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cartpolev0-cartpolev1

This commit is contained in:
xiaoyisd 2021-03-12 16:37:27 +08:00
parent 654771df13
commit 3d1ce87d70
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/rl/dqn/eval.py
View File

@ -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)

4
model_zoo/official/rl/dqn/train.py
View File

@ -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: