ch_RepSVTR_rec.yaml 916 B

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: null
  20. resume_path: null
  21. log_interval: 20
  22. eval_interval: 1
  23. save_interval: 1
  24. Evaluate:
  25. weight_path: "output/best_accuracy.pdparams"
  26. log_interval: 1
  27. Export:
  28. weight_path: https://paddleocr.bj.bcebos.com/pretrained/ch_SVTRv2_rec_mobile_trained.pdparams
  29. Predict:
  30. model_dir: "output/best_accuracy"
  31. input_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png"
  32. kernel_option:
  33. run_mode: paddle
  34. batch_size: 1