c6414c2a14cdf74addceeafeea55ee782e8cd391,main.py,,get_best_checkpoint_path,#Any#Any#Any#,266

Before Change

    checkpoint = []
    if phase == "target_train":
        assert task_name is not None, "Specify a task checkpoint to evaluate from."
        checkpoint = glob.glob(
            os.path.join(run_dir, task_name, "model_state_%s_epoch_*.best.th" % phase)
        )
    if len(checkpoint) == 0:
        checkpoint = glob.glob(os.path.join(run_dir, "model_state_untrained_pre_target_train.th"))
    if len(checkpoint) == 0 and phase == "pretrain":
        checkpoint = glob.glob(os.path.join(run_dir, "model_state_pretrain_epoch_*.best.th"))
    if len(checkpoint) > 0:
        assert_for_log(len(checkpoint) == 1, "Too many best checkpoints. Something is wrong.")

After Change

    checkpoint = []
    if phase == "target_train":
        if args.load_target_train_checkpoint not in ("none", ""):
            checkpoint = glob.glob(args.load_target_train_checkpoint)
            assert len(checkpoint) > 0, (
                "Specified load_target_train_checkpoint not found: %r"
                % args.load_target_train_checkpoint
            )

Instances