You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.5 KiB
53 lines
1.5 KiB
2 years ago
|
===========================train_params===========================
|
||
|
|
model_name:seg:fast_scnn
|
||
|
|
python:python
|
||
|
|
gpu_list:0|0,1
|
||
|
|
use_gpu:null|null
|
||
|
|
--precision:null
|
||
|
|
--num_epochs:lite_train_lite_infer=3|lite_train_whole_infer=3|whole_train_whole_infer=20
|
||
|
|
--save_dir:adaptive
|
||
|
|
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=4
|
||
|
|
--model_path:null
|
||
|
|
--config:lite_train_lite_infer=./test_tipc/configs/seg/fast_scnn/fast_scnn_rsseg.yaml|lite_train_whole_infer=./test_tipc/configs/seg/fast_scnn/fast_scnn_rsseg.yaml|whole_train_whole_infer=./test_tipc/configs/seg/fast_scnn/fast_scnn_rsseg.yaml
|
||
|
|
train_model_name:best_model
|
||
|
|
null:null
|
||
|
|
##
|
||
|
|
trainer:norm
|
||
|
|
norm_train:test_tipc/run_task.py train seg
|
||
|
|
pact_train:null
|
||
|
|
fpgm_train:null
|
||
|
|
distill_train:null
|
||
|
|
null:null
|
||
|
|
null:null
|
||
|
|
##
|
||
|
|
===========================eval_params===========================
|
||
|
|
eval:null
|
||
|
|
null:null
|
||
|
|
##
|
||
|
|
===========================export_params===========================
|
||
|
|
--save_dir:adaptive
|
||
|
|
--model_dir:adaptive
|
||
|
|
--fixed_input_shape:[-1,10,512,512]
|
||
|
|
norm_export:deploy/export/export_model.py
|
||
|
|
quant_export:null
|
||
|
|
fpgm_export:null
|
||
|
|
distill_export:null
|
||
|
|
export1:null
|
||
|
|
export2:null
|
||
|
|
===========================infer_params===========================
|
||
|
|
infer_model:null
|
||
|
|
infer_export:null
|
||
|
|
infer_quant:False
|
||
|
|
inference:test_tipc/infer.py
|
||
|
|
--device:cpu|gpu
|
||
|
|
--enable_mkldnn:True
|
||
|
|
--cpu_threads:6
|
||
|
|
--batch_size:1
|
||
|
|
--use_trt:False
|
||
|
|
--precision:fp32
|
||
|
|
--model_dir:null
|
||
|
|
--config:null
|
||
|
|
--save_log_path:null
|
||
|
|
--benchmark:True
|
||
|
|
--model_name:fast_scnn
|
||
|
|
null:null
|