3176941b5e9f96fd98e28b164f118bbe6ace6496,train.py,,,#,215

Before Change

    params = edict(getattr(prosr, args.model + "_params"))
    params.cmd = edict(vars(args))

    pprint(params)

    params.cmd.experiment_id = "{}_{}".format(args.model, args.name)
    checkpoint_dir = osp.join(CHECKPOINT_DIR, params.cmd.experiment_id)
    if not osp.isdir(checkpoint_dir):

After Change

            try:
                params = edict(yaml.load(stream))
            except yaml.YAMLError as exc:
                print(exc)
                sys.exit(0)
    else:
        params = edict(getattr(prosr, args.model + "_params"))

Instances