()
| 22 | |
| 23 | |
| 24 | def get_args(): |
| 25 | # Get some basic command line arguements |
| 26 | parser = argparse.ArgumentParser() |
| 27 | parser.add_argument('-e', '--env', help='environment ID', default='BreakoutNoFrameskip-v4') |
| 28 | parser.add_argument('-s', '--steps', help='training steps', type=int, default=int(80e6)) |
| 29 | parser.add_argument('--nenv', help='No. of environments', type=int, default=16) |
| 30 | return parser.parse_args() |
| 31 | |
| 32 | |
| 33 | def train(env_id, num_timesteps, num_cpu): |