train_infer_python.txt 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. ===========================train_params===========================
  2. model_name:seg:fast_scnn
  3. python:python
  4. gpu_list:0|0,1
  5. use_gpu:null|null
  6. --precision:null
  7. --num_epochs:lite_train_lite_infer=3|lite_train_whole_infer=3|whole_train_whole_infer=20
  8. --save_dir:adaptive
  9. --train_batch_size:lite_train_lite_infer=4|lite_train_whole_infer=4|whole_train_whole_infer=4
  10. --model_path:null
  11. --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
  12. train_model_name:best_model
  13. null:null
  14. ##
  15. trainer:norm
  16. norm_train:test_tipc/run_task.py train seg
  17. pact_train:null
  18. fpgm_train:null
  19. distill_train:null
  20. null:null
  21. null:null
  22. ##
  23. ===========================eval_params===========================
  24. eval:null
  25. null:null
  26. ##
  27. ===========================export_params===========================
  28. --save_dir:adaptive
  29. --model_dir:adaptive
  30. --fixed_input_shape:[-1,3,512,512]
  31. norm_export:deploy/export/export_model.py
  32. quant_export:null
  33. fpgm_export:null
  34. distill_export:null
  35. export1:null
  36. export2:null
  37. ===========================infer_params===========================
  38. infer_model:null
  39. infer_export:null
  40. infer_quant:False
  41. inference:test_tipc/infer.py
  42. --device:cpu|gpu
  43. --enable_mkldnn:True
  44. --cpu_threads:6
  45. --batch_size:1
  46. --use_trt:False
  47. --precision:fp32
  48. --model_dir:null
  49. --config:null
  50. --save_log_path:null
  51. --benchmark:True
  52. --model_name:fast_scnn
  53. null:null