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.
|
|
|
===========================train_params===========================
|
|
|
|
model_name:cd:bit
|
|
|
|
python:python
|
|
|
|
gpu_list:0|0,1
|
|
|
|
use_gpu:null|null
|
|
|
|
--precision:null
|
|
|
|
--num_epochs:lite_train_lite_infer=5|lite_train_whole_infer=5|whole_train_whole_infer=10
|
|
|
|
--save_dir:adaptive
|
|
|
|
--train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=8
|
|
|
|
--model_path:null
|
|
|
|
--config:lite_train_lite_infer=./test_tipc/configs/cd/bit/bit_airchange.yaml|lite_train_whole_infer=./test_tipc/configs/cd/bit/bit_airchange.yaml|whole_train_whole_infer=./test_tipc/configs/cd/bit/bit_levircd.yaml
|
|
|
|
train_model_name:best_model
|
|
|
|
null:null
|
|
|
|
##
|
|
|
|
trainer:norm
|
|
|
|
norm_train:test_tipc/run_task.py train cd
|
|
|
|
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,3,256,256]
|
|
|
|
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:bit
|
|
|
|
null:null
|