From 0034314376648460effc46223647963866ab9cab Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Wed, 8 Feb 2023 07:19:26 +0000 Subject: [PATCH] support to train swin with dynamic2static --- ...inTransformer_tiny_patch4_window7_224_train_infer_python.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 1d62a0c9..4d2caee3 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 @@ -54,7 +54,7 @@ null:null batch_size:104|128 fp_items:fp32|fp16 epoch:1 -model_type:norm_train +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========================== -- GitLab