From e9849cbdf6aebfe823fd5889242cad8bbdeacb71 Mon Sep 17 00:00:00 2001
From: marsjacobs <minsoo2333@hanyang.ac.kr>
Date: Sat, 8 Apr 2023 12:03:06 +0900
Subject: [PATCH] qat option fixed

---
 run_TI_step_1.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/run_TI_step_1.sh b/run_TI_step_1.sh
index a2f84bd..18c6df8 100644
--- 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} 
\ No newline at end of file
+--learning_rate ${learning_rate}