Skip to content

Commit

Permalink
[Fix] add stdc to tipc train benchmark (#2954)
Browse files Browse the repository at this point in the history
  • Loading branch information
juncaipeng authored Feb 4, 2023
1 parent 2f5798e commit f8e8ba7
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions test_tipc/configs/stdc_stdc1/train_infer_python.txt
Original file line number Diff line number Diff line change
Expand Up @@ -13,15 +13,15 @@ train_infer_img_dir:test_tipc/data/cityscapes/cityscapes_val_5.list
null:null
##
trainer:norm_train
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
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
pact_train:null
fpgm_train:null
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:null
eval:tools/val.py --config test_tipc/configs/stdc_stdc1/stdc1_seg_cityscapes_1024x512_80k.yml --device gpu --num_workers 8
null:null
##
===========================export_params===========================
Expand Down Expand Up @@ -52,3 +52,12 @@ inference:deploy/python/infer.py
--model_name:stdc_stdc1
===========================infer_benchmark_params==========================
random_infer_input:[{float32,[3,1024,2048]}]
===========================to_static_train_benchmark_params===========================
to_static_train:--opts to_static_training=True
===========================train_benchmark_params==========================
batch_size:2|4
fp_items:fp32|fp16
epoch:800
--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;FLAGS_cudnn_deterministic=False
log_iters:15;skip_iters:2;repeats:500

0 comments on commit f8e8ba7

Please sign in to comment.