Skip to content

Commit f8e8ba7

Browse files
authored
[Fix] add stdc to tipc train benchmark (#2954)
1 parent 2f5798e commit f8e8ba7

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

test_tipc/configs/stdc_stdc1/train_infer_python.txt

+11-2
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,15 @@ train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list
1313
null:null
1414
##
1515
trainer:norm_train
16-
norm_train:tools/train.py --config test_tipc/configs/stdc_stdc1/stdc1_seg_cityscapes_1024x512_80k.yml --device gpu --do_eval --save_interval 500 --seed 100
16+
norm_train:tools/train.py --config test_tipc/configs/stdc_stdc1/stdc1_seg_cityscapes_1024x512_80k.yml --device gpu --save_interval 500 --seed 100 --num_workers 8
1717
pact_train:null
1818
fpgm_train:null
1919
distill_train:null
2020
null:null
2121
null:null
2222
##
2323
===========================eval_params===========================
24-
eval:null
24+
eval:tools/val.py --config test_tipc/configs/stdc_stdc1/stdc1_seg_cityscapes_1024x512_80k.yml --device gpu --num_workers 8
2525
null:null
2626
##
2727
===========================export_params===========================
@@ -52,3 +52,12 @@ inference:deploy/python/infer.py
5252
--model_name:stdc_stdc1
5353
===========================infer_benchmark_params==========================
5454
random_infer_input:[{float32,[3,1024,2048]}]
55+
===========================to_static_train_benchmark_params===========================
56+
to_static_train:--opts to_static_training=True
57+
===========================train_benchmark_params==========================
58+
batch_size:2|4
59+
fp_items:fp32|fp16
60+
epoch:800
61+
--profiler_options:'batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile'
62+
flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096;FLAGS_cudnn_deterministic=False
63+
log_iters:15;skip_iters:2;repeats:500

0 commit comments

Comments
 (0)