d9f194f797f406969f454ba7338c798c14cff01e,baselines/a2c/run_atari.py,,main,#,36

Before Change

def main():
    train("BreakoutNoFrameskip-v4", num_timesteps=int(40e6), seed=0, policy="cnn", lrschedule="linear", num_cpu=16)


if __name__ == "__main__":
    main()

After Change

def main():
    import argparse
    parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument("--env", help="environment ID", default="BreakoutNoFrameskip-v4")
    parser.add_argument("--seed", help="RNG seed", type=int, default=0)
    parser.add_argument("--policy", help="Policy architecture", choices=["cnn", "lstm", "lnlstm"], default="cnn")
    parser.add_argument("--lrschedule", help="Learning rate schedule", choices=["constant", "linear"], default="constant")
    parser.add_argument("--million_frames", help="How many frames to train (/ 1e6). "
        "This number gets divided by 4 due to frameskip", type=int, default=40)
    args = parser.parse_args()
    train(args.env, num_frames=1e6 * args.million_frames, seed=args.seed, 
        policy=args.policy, lrschedule=args.lrschedule, num_cpu=16)

if __name__ == "__main__":
    main()

Instances