tinypose_128x96_train_infer_python.txt 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960
  1. ===========================train_params===========================
  2. model_name:tinypose_128x96
  3. python:python3.7
  4. gpu_list:0|0,1
  5. use_gpu:True
  6. auto_cast:null
  7. epoch:lite_train_lite_infer=1|lite_train_whole_infer=1|whole_train_whole_infer=420
  8. save_dir:null
  9. TrainReader.batch_size:lite_train_lite_infer=2|lite_train_whole_infer=2|whole_train_whole_infer=512
  10. pretrain_weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams
  11. trained_model_name:model_final.pdparams
  12. train_infer_img_dir:./dataset/coco/test2017/
  13. filename:null
  14. ##
  15. trainer:norm_train
  16. norm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o
  17. pact_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o
  18. fpgm_train:tools/train.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o
  19. distill_train:null
  20. null:null
  21. null:null
  22. ##
  23. ===========================eval_params===========================
  24. eval:tools/eval.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o
  25. null:null
  26. ##
  27. ===========================infer_params===========================
  28. --output_dir:./output_inference
  29. weights:https://paddledet.bj.bcebos.com/models/keypoint/tinypose_128x96.pdparams
  30. norm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml -o
  31. pact_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_pact -o
  32. fpgm_export:tools/export_model.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config _template_fpgm -o
  33. distill_export:null
  34. export1:null
  35. export2:null
  36. kl_quant_export:tools/post_quant.py -c test_tipc/configs/keypoint/tinypose_128x96.yml --slim_config configs/slim/post_quant/tinypose_128x96_ptq.yml -o
  37. ##
  38. infer_mode:norm|kl_quant
  39. infer_quant:False|True
  40. inference:./deploy/python/keypoint_infer.py
  41. --device:gpu|cpu
  42. --enable_mkldnn:False
  43. --cpu_threads:4
  44. --batch_size:1|2
  45. --use_tensorrt:null
  46. --run_mode:paddle
  47. --model_dir:
  48. --image_dir:./dataset/coco/test2017/
  49. --save_log_path:null
  50. --run_benchmark:False
  51. null:null
  52. ===========================disable_train_benchmark==========================
  53. batch_size:512
  54. fp_items:fp32|fp16
  55. epoch:1
  56. repeat:1
  57. --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
  58. flags:null
  59. ===========================infer_benchmark_params===========================
  60. random_infer_input:[{float32,[3,128,96]}]