diff --git a/test_tipc/configs/ConvNeXt/ConvNeXt_small_train_infer_python.txt b/test_tipc/configs/ConvNeXt/ConvNeXt_small_train_infer_python.txt index d3f5896ef6809994622c7dd9ef017340c29683b0..7a6281eb6a07ba096599c8cca27e13caeedc4202 100644 --- a/test_tipc/configs/ConvNeXt/ConvNeXt_small_train_infer_python.txt +++ b/test_tipc/configs/ConvNeXt/ConvNeXt_small_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_small.yaml pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== diff --git a/test_tipc/configs/HRNet/HRNet_W18_C_train_infer_python.txt b/test_tipc/configs/HRNet/HRNet_W18_C_train_infer_python.txt index a4f88a4c3c0a991acad68c2764afbd74e176ea86..02b19d5aeb48bf4d9d93be9454b099ebae35d0bd 100644 --- a/test_tipc/configs/HRNet/HRNet_W18_C_train_infer_python.txt +++ b/test_tipc/configs/HRNet/HRNet_W18_C_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/HRNet/HRNet_W18_C.yaml -o Gl pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== diff --git a/test_tipc/configs/HRNet/HRNet_W40_C_train_infer_python.txt b/test_tipc/configs/HRNet/HRNet_W40_C_train_infer_python.txt index e45fe37f3631ce054a6def0cb657dfa93551b91b..31df02f8ab16f4ecca2edd7a66eae5080c1dd4d1 100644 --- a/test_tipc/configs/HRNet/HRNet_W40_C_train_infer_python.txt +++ b/test_tipc/configs/HRNet/HRNet_W40_C_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/HRNet/HRNet_W40_C.yaml -o Gl pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== @@ -58,4 +58,4 @@ model_type:norm_train|to_static_train --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}] diff --git a/test_tipc/configs/ResNeSt/ResNeSt101_train_infer_python.txt b/test_tipc/configs/ResNeSt/ResNeSt101_train_infer_python.txt index e41267932c64094775e389e9418e7a45932d673f..9f000d2a7e1bf34465270221e6f24cefbeeee6bb 100644 --- a/test_tipc/configs/ResNeSt/ResNeSt101_train_infer_python.txt +++ b/test_tipc/configs/ResNeSt/ResNeSt101_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/ResNeSt/ResNeSt101.yaml -o G pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== @@ -58,4 +58,4 @@ model_type:norm_train|to_static_train --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}]