Merge pull request #1555 from RainFrost1/batch_size

change large model batch_size for tipc

test_tipc/config/RedNet/RedNet152_train_infer_python.txt

@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val

test_tipc/config/Twins/alt_gvt_large_train_infer_python.txt

@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val

test_tipc/config/Twins/pcpvt_large_train_infer_python.txt

@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val

test_tipc/config/VisionTransformer/ViT_large_patch16_224_train_infer_python.txt

@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val

test_tipc/config/VisionTransformer/ViT_large_patch16_384_train_infer_python.txt

@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val

test_tipc/config/VisionTransformer/ViT_large_patch32_384_train_infer_python.txt

@@ -6,7 +6,7 @@ gpu_list:0|0,1
 -o Global.auto_cast:null
 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
 -o Global.output_dir:./output/
--o DataLoader.Train.sampler.batch_size:8
+-o DataLoader.Train.sampler.batch_size:2
 -o Global.pretrained_model:null
 train_model_name:latest
 train_infer_img_dir:./dataset/ILSVRC2012/val

test_tipc/test_train_inference_python.sh

@@ -146,7 +146,6 @@ function func_inference(){
                         last_status=${PIPESTATUS[0]}
                         eval "cat ${_save_log_path}"
                         status_check $last_status "${command}" "../${status_log}"
-                        
                     done
                 done
             done
@@ -289,8 +288,10 @@ else
                 # run train
 		eval "unset CUDA_VISIBLE_DEVICES"
 		export FLAGS_cudnn_deterministic=True
+		sleep 5
                 eval $cmd
                 status_check $? "${cmd}" "${status_log}"
+                sleep 5
 		
 		if [[ $FILENAME == *GeneralRecognition* ]]; then
 		    set_eval_pretrain=$(func_set_params "${pretrain_model_key}" "${save_log}/RecModel/${train_model_name}")
@@ -307,6 +308,7 @@ else
                     eval_cmd="${python} ${eval_py} ${set_eval_pretrain} ${set_use_gpu} ${set_eval_params1}" 
                     eval $eval_cmd
                     status_check $? "${eval_cmd}" "${status_log}"
+                    sleep 5
                 fi
                 # run export model
                 if [ ${run_export} != "null" ]; then