qat option fixed

MarsJacobs · Apr 8, 2023 · e9849cb · e9849cb
1 parent 44213f4
commit e9849cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run_TI_step_1.sh b/run_TI_step_1.sh
@@ -35,7 +35,7 @@ teacher_stochastic=0
 teacher_inverted=0
 
 # Training Type (downstream, qat_normal, qat_step1, qat_step2)
-training_type=qat_normal
+training_type=qat_step1
 
 # DA Options
 aug_train=0
@@ -56,4 +56,4 @@ CUDA_VISIBLE_DEVICES=$1 python /home/ms/workspace/git/Teacher-Intervention-KD-QA
 --neptune ${neptune} \
 --aug_N ${aug_N} \
 --num_train_epochs 3 --seed 1 \
---learning_rate ${learning_rate} 
+--learning_rate ${learning_rate}