diff --git a/official/cv/ctpn/scripts/run_standalone_train_gpu.sh b/official/cv/ctpn/scripts/run_standalone_train_gpu.sh
index 757d70c6f165c406a26b49d1a7cbc8172b6c124a..ee3f1cde42db2d28f01429919933508c6e18e3cf 100644
--- a/official/cv/ctpn/scripts/run_standalone_train_gpu.sh
+++ b/official/cv/ctpn/scripts/run_standalone_train_gpu.sh
@@ -52,12 +52,11 @@ cp -r ./src ./train
 cd ./train || exit
 
 export DEVICE_NUM=1
-export DEVICE_ID=$3
 export RANK_ID=0
 export RANK_SIZE=1
 
-echo "start training for device $DEVICE_ID"
-export CUDA_VISIBLE_DEVICES=$DEVICE_ID
+echo "start training for device $3"
+export CUDA_VISIBLE_DEVICES=$3
 env > env.log
 python train.py --task_type=$TASK_TYPE --pre_trained=$PRETRAINED_PATH --device_target="GPU" &> log &
 cd ..
diff --git a/official/nlp/tinybert/src/utils.py b/official/nlp/tinybert/src/utils.py
index a8a3213cc6cf26ec220f037ba22bc0bfee037752..d614d30c23a1c66ea0f475b1926c5d26245a9e86 100644
--- a/official/nlp/tinybert/src/utils.py
+++ b/official/nlp/tinybert/src/utils.py
@@ -98,6 +98,7 @@ class EvalCallBack(Callback):
             input_ids, input_mask, token_type_id, label_ids = input_data
             self.network.set_train(False)
             logits = self.network(input_ids, token_type_id, input_mask)
+            self.network.set_train(True)
             callback.update(logits, label_ids)
         acc = callback.acc_num / callback.total_num
         with open("./eval.log", "a+") as f: