diff --git a/test_tipc/configs/HRNet/HRNet_W48_C_train_infer_python.txt b/test_tipc/configs/HRNet/HRNet_W48_C_train_infer_python.txt index c8c855701cb42b38be166f438b43f6efac9ff06b..474465b68fcd75323a0214a73521b0da0ec1bc8c 100644 --- a/test_tipc/configs/HRNet/HRNet_W48_C_train_infer_python.txt +++ b/test_tipc/configs/HRNet/HRNet_W48_C_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:64|128 +batch_size:64 fp_items:fp32 epoch:1 model_type:norm_train diff --git a/test_tipc/configs/MobileNetV1/MobileNetV1_train_infer_python.txt b/test_tipc/configs/MobileNetV1/MobileNetV1_train_infer_python.txt index 585f719c7dfa71c63187b0f81ea1f0783c809417..1b6a696394c0e2a321b8d92b439e205245b812a6 100644 --- a/test_tipc/configs/MobileNetV1/MobileNetV1_train_infer_python.txt +++ b/test_tipc/configs/MobileNetV1/MobileNetV1_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:64|128 +batch_size:64 fp_items:fp32|fp16 epoch:1 model_type:norm_train diff --git a/test_tipc/configs/MobileNetV2/MobileNetV2_train_infer_python.txt b/test_tipc/configs/MobileNetV2/MobileNetV2_train_infer_python.txt index c9eaf391ab60d2d8fa513ab89bfd12fc27859f53..dd78b28959e5bd3a1ac0ef06997eea3e52d58ce0 100644 --- a/test_tipc/configs/MobileNetV2/MobileNetV2_train_infer_python.txt +++ b/test_tipc/configs/MobileNetV2/MobileNetV2_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:64|128 +batch_size:64 fp_items:fp32|fp16 epoch:1 model_type:norm_train|to_static_train diff --git a/test_tipc/configs/MobileNetV3/MobileNetV3_large_x1_0_train_infer_python.txt b/test_tipc/configs/MobileNetV3/MobileNetV3_large_x1_0_train_infer_python.txt index 232357baa19f20498433bec8b50cb2177eb030c1..7c2c7c553caacaf37cf0682b3f98cb17334ca0af 100644 --- a/test_tipc/configs/MobileNetV3/MobileNetV3_large_x1_0_train_infer_python.txt +++ b/test_tipc/configs/MobileNetV3/MobileNetV3_large_x1_0_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:256|640 +batch_size:256 fp_items:fp32|fp16 epoch:1 model_type:norm_train|to_static_train diff --git a/test_tipc/configs/ResNet/ResNet152_train_infer_python.txt b/test_tipc/configs/ResNet/ResNet152_train_infer_python.txt index 3e4d2d34a523c0c9a804905892f49193ca74f8d3..3bef788c5ec2ac63af2d789688acfc68e18a4efd 100644 --- a/test_tipc/configs/ResNet/ResNet152_train_infer_python.txt +++ b/test_tipc/configs/ResNet/ResNet152_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:32|64 +batch_size:64 fp_items:fp32|fp16 epoch:1 model_type:norm_train|to_static_train diff --git a/test_tipc/configs/ResNet/ResNet50_train_infer_python.txt b/test_tipc/configs/ResNet/ResNet50_train_infer_python.txt index 4670aa3eab0c8b7245041304ef280e936ae03d0f..145579cf7077db96afcf2ce91553c68847344f7d 100644 --- a/test_tipc/configs/ResNet/ResNet50_train_infer_python.txt +++ b/test_tipc/configs/ResNet/ResNet50_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:256|128|64 +batch_size:256|64 fp_items:fp32|fp16 epoch:1 model_type:norm_train|to_static_train diff --git a/test_tipc/configs/ShuffleNet/ShuffleNetV2_x1_0_train_infer_python.txt b/test_tipc/configs/ShuffleNet/ShuffleNetV2_x1_0_train_infer_python.txt index 61cdd46b7eb8624438db391aa4d6019c6a265d6d..d89bfe27e12907d038dccc8c20309f17c2c070ec 100644 --- a/test_tipc/configs/ShuffleNet/ShuffleNetV2_x1_0_train_infer_python.txt +++ b/test_tipc/configs/ShuffleNet/ShuffleNetV2_x1_0_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:256|1536 +batch_size:256 fp_items:fp32|fp16 epoch:2 model_type:norm_train diff --git a/test_tipc/configs/SwinTransformer/SwinTransformer_tiny_patch4_window7_224_train_infer_python.txt b/test_tipc/configs/SwinTransformer/SwinTransformer_tiny_patch4_window7_224_train_infer_python.txt index 5c5c594718dcb4789c3b6e1436c4b5d1bf6d72fe..f72a233ee7b25eada2938aceeca25ff2d16d4339 100644 --- a/test_tipc/configs/SwinTransformer/SwinTransformer_tiny_patch4_window7_224_train_infer_python.txt +++ b/test_tipc/configs/SwinTransformer/SwinTransformer_tiny_patch4_window7_224_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:104|128 +batch_size:128 fp_items:fp32|fp16 epoch:1 model_type:norm_train|to_static_train diff --git a/test_tipc/configs/Twins/alt_gvt_base_train_infer_python.txt b/test_tipc/configs/Twins/alt_gvt_base_train_infer_python.txt index a39db9dc84c5896d15f63e5e1f98518eb5fed3e9..f99ff28177b69c65fd84d6b5d13498447e7bf6e6 100644 --- a/test_tipc/configs/Twins/alt_gvt_base_train_infer_python.txt +++ b/test_tipc/configs/Twins/alt_gvt_base_train_infer_python.txt @@ -51,7 +51,7 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml null:null null:null ===========================train_benchmark_params========================== -batch_size:64|144 +batch_size:128 fp_items:fp32 epoch:1 model_type:norm_train