diff --git a/test_tipc/config/RedNet/RedNet152_train_infer_python.txt b/test_tipc/config/RedNet/RedNet152_train_infer_python.txt index dcedf613811b0d911106ad2776a849d552253576..82e8189d5b69858bc70f93cbfd9d22d2d9177b65 100644 --- a/test_tipc/config/RedNet/RedNet152_train_infer_python.txt +++ b/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 diff --git a/test_tipc/config/Twins/alt_gvt_large_train_infer_python.txt b/test_tipc/config/Twins/alt_gvt_large_train_infer_python.txt index fb40a05c89c2cef1e02ab0d474ea12f31a7006c6..d75f8a14cba0cf347f6219cf288a8bbc903a8a32 100644 --- a/test_tipc/config/Twins/alt_gvt_large_train_infer_python.txt +++ b/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 diff --git a/test_tipc/config/Twins/pcpvt_large_train_infer_python.txt b/test_tipc/config/Twins/pcpvt_large_train_infer_python.txt index 249fcfe36a6721aade1284d9f04aae0c4c89efad..928c0b30db0dbc0e33108b7ee50cf91ddd84f821 100644 --- a/test_tipc/config/Twins/pcpvt_large_train_infer_python.txt +++ b/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 diff --git a/test_tipc/config/VisionTransformer/ViT_large_patch16_224_train_infer_python.txt b/test_tipc/config/VisionTransformer/ViT_large_patch16_224_train_infer_python.txt index 80b92a3149caf13c22381d1f6b777717b0668b52..ec77af6defbc0c062bac5517227890ea9f168909 100644 --- a/test_tipc/config/VisionTransformer/ViT_large_patch16_224_train_infer_python.txt +++ b/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 diff --git a/test_tipc/config/VisionTransformer/ViT_large_patch16_384_train_infer_python.txt b/test_tipc/config/VisionTransformer/ViT_large_patch16_384_train_infer_python.txt index dd067233703248d8a73feeaa8ea4fa40f6f61e70..e05c6df96bd61f67a4bf5e08e76f8cd5d334ddd5 100644 --- a/test_tipc/config/VisionTransformer/ViT_large_patch16_384_train_infer_python.txt +++ b/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 diff --git a/test_tipc/config/VisionTransformer/ViT_large_patch32_384_train_infer_python.txt b/test_tipc/config/VisionTransformer/ViT_large_patch32_384_train_infer_python.txt index b5211889a2e91c7f03ebe7f7539327d0007ac4ff..43b313f58c047b94af2a64d7460910530bb488cf 100644 --- a/test_tipc/config/VisionTransformer/ViT_large_patch32_384_train_infer_python.txt +++ b/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 diff --git a/test_tipc/test_train_inference_python.sh b/test_tipc/test_train_inference_python.sh index 23b2f70a18cf8b5dfb7c149e2c5821788d2f34fa..cecfa93e70d4767b43f667ab01134d03409458a0 100644 --- a/test_tipc/test_train_inference_python.sh +++ b/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