Merge pull request #1301 from RainFrost1/add_model

change infer batch_size to 1|16 for whole_train

tests/config/DarkNet53.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/GhostNet_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/GhostNet_x1_0.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/GhostNet_x1_3.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/HRNet_W18_C.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/LeViT_128S.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|Fasle
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV1.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV1_x0_25.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV1_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV1_x0_75.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV2.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV2_x0_25.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV2_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV2_x0_75.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV2_x1_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV2_x2_0.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_large_x0_35.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_large_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_large_x0_75.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_large_x1_0.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_large_x1_25.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_small_x0_35.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_small_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_small_x0_75.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_small_x1_0.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/MobileNetV3_small_x1_25.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/PPLCNet_x0_25.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/PPLCNet_x0_35.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/PPLCNet_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/PPLCNet_x1_0.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/ResNeXt101_vd_64x4d.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/ResNet50_vd.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/ShuffleNetV2_x0_25.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/ShuffleNetV2_x0_33.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/ShuffleNetV2_x0_5.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/ShuffleNetV2_x1_0.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference

tests/config/SwinTransformer_tiny_patch4_window7_224.txt

@@ -41,7 +41,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml
 -o Global.use_gpu:True|False
 -o Global.enable_mkldnn:True|False
 -o Global.cpu_num_threads:1|6
--o Global.batch_size:1
+-o Global.batch_size:1|16
 -o Global.use_tensorrt:True|False
 -o Global.use_fp16:True|False
 -o Global.inference_model_dir:../inference