yolov3_darknet53_270e_coco_params.txt 2.1 KB
Newer Older
1 2 3
===========================train_params===========================
model_name:yolov3_darknet53_270e_coco
python:python3.7
C
cnn 已提交
4 5
gpu_list:0
use_gpu:True
6
auto_cast:null|amp
S
shangliang Xu 已提交
7
epoch:lite_train_infer=1|whole_train_infer=270
C
cnn 已提交
8
save_dir:./output/
S
shangliang Xu 已提交
9 10
TrainReader.batch_size:lite_train_infer=2|whole_train_infer=8
pretrain_weights:null
C
cnn 已提交
11
train_model_name:yolov3_darknet53_270e_coco/model_final.pdparams
S
shangliang Xu 已提交
12 13
train_infer_img_dir:./tests/demo/
filename:yolov3_darknet53_270e_coco
14
##
S
shangliang Xu 已提交
15
trainer:norm_train|pact_train|fpgm_train
16
norm_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml -o
C
cnn 已提交
17 18
pact_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/quant/yolov3_darknet_qat.yml -o
fpgm_train:tools/train.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/prune/yolov3_darknet_prune_fpgm.yml -o
19 20 21 22 23 24 25 26 27
distill_train:null
null:null
null:null
##
===========================eval_params===========================
eval:tools/eval.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml -o
null:null
##
===========================infer_params===========================
S
shangliang Xu 已提交
28
--output_dir:null
C
cnn 已提交
29
weights:
30
norm_export:tools/export_model.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml -o
S
shangliang Xu 已提交
31 32
quant_export:tools/export_model.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/quant/yolov3_darknet_qat.yml -o
fpgm_export:tools/export_model.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/prune/yolov3_darknet_prune_fpgm.yml -o
33
distill_export:null
S
shangliang Xu 已提交
34
kl_quant:tools/post_quant.py -c configs/yolov3/yolov3_darknet53_270e_coco.yml --slim_config configs/slim/post_quant/yolov3_darknet53_ptq.yml -o
S
shangliang Xu 已提交
35
export2:null
36
##
S
shangliang Xu 已提交
37
infer_model:yolov3_darknet53_270e_coco.pdparams|yolov3_darknet_coco_qat.pdparams|yolov3_darknet_prune_fpgm.pdparams
S
shangliang Xu 已提交
38
infer_model_dir:./tests/weights
S
shangliang Xu 已提交
39
infer_quant:False|True|False
40
inference:deploy/python/infer.py
S
shangliang Xu 已提交
41
--device:gpu|cpu
42 43 44 45
--enable_mkldnn:False|True
--cpu_threads:1|4
--batch_size:1|2
--use_tensorrt:null
S
shangliang Xu 已提交
46 47 48
--run_mode:fluid|trt_fp32|trt_fp16|trt_int8
--model_dir:
--image_dir:./dataset/coco/val2017/
49 50 51
--save_log_path:null
--run_benchmark:True
null:null