RoboschoolReacher_v0_2017may.py 文件源码

def demo_run():
    env = gym.make("RoboschoolReacher-v1")
    pi = SmallReactivePolicy(env.observation_space, env.action_space)

    while 1:
        frame = 0
        score = 0
        obs = env.reset()

        while 1:
            a = pi.act(obs)
            obs, r, done, _ = env.step(a)
            score += r
            frame += 1
            still_open = env.render("human")
            if still_open==False:
                return
            if not done: continue
            print("score=%0.2f in %i frames" % (score, frame))
            break