ch_RepSVTR_rec.yaml 1.0 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. Global:
  2. model: ch_RepSVTR_rec
  3. mode: check_dataset # check_dataset/train/evaluate/predict
  4. dataset_dir: "/paddle/dataset/paddlex/ocr_rec/ocr_rec_dataset_examples"
  5. device: gpu:0,1,2,3
  6. output: "output"
  7. CheckDataset:
  8. convert:
  9. enable: False
  10. src_dataset_type: null
  11. split:
  12. enable: False
  13. train_percent: null
  14. val_percent: null
  15. Train:
  16. epochs_iters: 20
  17. batch_size: 8
  18. learning_rate: 0.001
  19. pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/ch_RepSVTR_rec_pretrained.pdparams
  20. resume_path: null
  21. log_interval: 20
  22. eval_interval: 1
  23. save_interval: 1
  24. Evaluate:
  25. weight_path: "output/best_accuracy/best_accuracy.pdparams"
  26. log_interval: 1
  27. Export:
  28. weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/ch_RepSVTR_rec_pretrained.pdparams
  29. Predict:
  30. batch_size: 1
  31. model_dir: "output/best_accuracy/inference"
  32. input: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png"
  33. kernel_option:
  34. run_mode: paddle