From 4a72afc4c1fc0864a2eb9f861724961ea14c7a6a Mon Sep 17 00:00:00 2001 From: zhangyubo0722 Date: Thu, 16 Mar 2023 12:35:45 +0000 Subject: [PATCH] add_tipc_config9 --- .../DeiT/DeiT_tiny_patch16_224_train_infer_python.txt | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt b/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt index b0d4efd1..fe42aa82 100644 --- a/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt +++ b/test_tipc/configs/DeiT/DeiT_tiny_patch16_224_train_infer_python.txt @@ -17,7 +17,7 @@ norm_train:tools/train.py -c ppcls/configs/ImageNet/DeiT/DeiT_tiny_patch16_224.y pact_train:null fpgm_train:null distill_train:null -null:null +to_static_train:-o Global.to_static=True null:null ## ===========================eval_params=========================== @@ -50,5 +50,12 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o Global.benchmark:False null:null null:null +===========================train_benchmark_params========================== +batch_size:256 +fp_items:fp32|fp16 +epoch:1 +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]}] -- GitLab