small fix transfer_analysis.sh (#946)

model_*.th might indicate several models; fixed to model_*.best.th
nyu-mll · Oct 27, 2019 · c07efbb · c07efbb
1 parent dc4f69d
commit c07efbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/taskmaster/transfer_analysis.sh b/scripts/taskmaster/transfer_analysis.sh
@@ -35,7 +35,7 @@ function run_intermediate_to_target_task() {
     # This function can also be used to finetune on a probing task as well. 
     # Usage: run_intermediate_to_target_task <intemeidate_task> <target_task> <directory_to_project_dir>
     OVERRIDES="exp_name=$1, run_name=$2"
-    OVERRIDES+=", target_tasks=$2, load_model=1, load_target_train_checkpoint=$3/roberta-large/$1/$1/model_*.th, pretrain_tasks=\"\","
+    OVERRIDES+=", target_tasks=$2, load_model=1, load_target_train_checkpoint=$3/roberta-large/$1/$1/model_*.best.th, pretrain_tasks=\"\","
     OVERRIDES+="input_module=roberta-large,"
     OVERRIDES+="do_pretrain=0, do_target_task_training=1"
     run_exp "jiant/config/taskmaster/base_roberta.conf" "${OVERRIDES}"