diff --git a/test_tipc/configs/ESNet/ESNet_x1_0_train_infer_python.txt b/test_tipc/configs/ESNet/ESNet_x1_0_train_infer_python.txt index 6b3787518e4a79eb5432b15a800223fa22a03593..00d83aa2f6daa6b3af75d65656ae20dffae74a70 100644 --- a/test_tipc/configs/ESNet/ESNet_x1_0_train_infer_python.txt +++ b/test_tipc/configs/ESNet/ESNet_x1_0_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/ESNet/ESNet_x1_0.yaml -o Glo 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/TNT/TNT_small_train_infer_python.txt b/test_tipc/configs/TNT/TNT_small_train_infer_python.txt index 5d8d20cd6c31a98e85d389792e1dfdb601b35b97..b7c36aea05eb7bf4612ef09a506ee81f4d614496 100644 --- a/test_tipc/configs/TNT/TNT_small_train_infer_python.txt +++ b/test_tipc/configs/TNT/TNT_small_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/TNT/TNT_small.yaml -o Global 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]}]