modify tinybert and ctpn scripts

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 ..

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: