Attempted fix.
@@ -498,7 +498,7 @@ def main(cl_arguments):
pre_target_train = get_best_checkpoint_path(args.run_dir, "pretrain")
load_model_state(
- model, pre_finetune_path, args.cuda, skip_task_models=[], strict=strict
+ model, pre_target_train, args.cuda, skip_task_models=[], strict=strict
)
else: # args.transfer_paradigm == "frozen":
# Load the current overall best model.
