add profile

d5989a82 · LDOUBLEV · 8f5e2f3d · d5989a82
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt +1 -1

未找到文件。
--- a/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
+++ b/test_tipc/configs/det_mv3_db_v2.0/train_benchmark.txt
@@ -13,7 +13,7 @@ train_infer_img_dir:null
 null:null
 ##
 trainer:norm_train
-norm_train:tools/train.py -c configs/det/det_mv3_db.yml -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained
+norm_train:tools/train.py -c configs/det/det_mv3_db.yml --profiler_options="batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile" -o Global.pretrained_model=./pretrain_models/MobileNetV3_large_x0_5_pretrained
 pact_train:null
 fpgm_train:null
 distill_train:null