train_infer_python.txt 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. ===========================train_params===========================
  2. model_name:vi_layoutxlm_ser
  3. python:python3.7
  4. gpu_list:0|0,1
  5. Global.use_gpu:True|True
  6. Global.auto_cast:fp32
  7. Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=17
  8. Global.save_model_dir:./output/
  9. Train.loader.batch_size_per_card:lite_train_lite_infer=4|whole_train_whole_infer=8
  10. Architecture.Backbone.checkpoints:null
  11. train_model_name:latest
  12. train_infer_img_dir:ppstructure/docs/kie/input/zh_val_42.jpg
  13. null:null
  14. ##
  15. trainer:norm_train
  16. norm_train:tools/train.py -c ./configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh.yml -o Global.print_batch_step=1 Global.eval_batch_step=[1000,1000] Train.loader.shuffle=false
  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. ===========================infer_params===========================
  28. Global.save_inference_dir:./output/
  29. Architecture.Backbone.checkpoints:
  30. norm_export:tools/export_model.py -c ./configs/kie/vi_layoutxlm/ser_vi_layoutxlm_xfund_zh.yml -o
  31. quant_export:
  32. fpgm_export:
  33. distill_export:null
  34. export1:null
  35. export2:null
  36. ##
  37. infer_model:null
  38. infer_export:null
  39. infer_quant:False
  40. inference:ppstructure/kie/predict_kie_token_ser.py --kie_algorithm=LayoutXLM --ser_dict_path=train_data/XFUND/class_list_xfun.txt --output=output --ocr_order_method=tb-yx
  41. --use_gpu:True|False
  42. --enable_mkldnn:False
  43. --cpu_threads:6
  44. --rec_batch_num:1
  45. --use_tensorrt:False
  46. --precision:fp32
  47. --ser_model_dir:
  48. --image_dir:./ppstructure/docs/kie/input/zh_val_42.jpg
  49. null:null
  50. --benchmark:False
  51. null:null
  52. ===========================infer_benchmark_params==========================
  53. random_infer_input:[{float32,[3,224,224]}]
  54. ===========================train_benchmark_params==========================
  55. batch_size:4
  56. fp_items:fp32|fp16
  57. epoch:3
  58. --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile
  59. flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98