tinypose_128x96_train_infer_python.txt 2.0 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
===========================train_params===========================
model_name:tinypose_128x96
python:python3.7
gpu_list:0|0,1
use_gpu:True
auto_cast:null
epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=420
save_dir:null
TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=512
pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams
trained_model_name:model_final.pdparams
train_infer_img_dir:./dataset/coco/test2017/
filename:null
##
trainer:norm_train
16 17 18
norm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o
pact_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o
fpgm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o
19 20 21 22 23
distill_train:null
null:null
null:null
##
===========================eval_params===========================
24
eval:tools/eval.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o
25 26 27 28 29
null:null
##
===========================infer_params===========================
--output_dir:./output_inference
weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams
30 31 32
norm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o
pact_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o
fpgm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o
33 34 35
distill_export:null
export1:null
export2:null
36
kl_quant_export:tools/post_quant.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_kl_quant -o
37 38 39
##
infer_mode:norm
infer_quant:False
40
inference:./deploy/python/keypoint_infer.py
41 42 43
--device:gpu|cpu
--enable_mkldnn:True|False
--cpu_threads:1|6
44
--batch_size:1|2
45 46 47 48 49 50 51
--use_tensorrt:null
--run_mode:fluid|trt_fp32|trt_fp16
--model_dir:
--image_dir:./dataset/coco/test2017/
--save_log_path:null
--run_benchmark:True
null:null