train.py 文件源码

def run():
    args = parser.parse_args()
    cmds, notes = create_commands(
        "a3c",
        args.num_workers,
        args.remotes,
        args.env_id,
        args.log_dir,
        mode=args.mode,
        hparams=hparams.get_hparams(args, ignore_default=True))
    if args.dry_run:
        print("Dry-run mode due to -n flag, otherwise the following commands would be executed:")
    else:
        print("Executing the following commands:")
    print("\n".join(cmds))
    print("")
    if not args.dry_run:
        if args.mode == "tmux":
            os.environ["TMUX"] = ""
        os.system("\n".join(cmds))
    print('\n'.join(notes))