SLANet.yaml 931 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. Global:
  2. model: SLANet
  3. mode: check_dataset # check_dataset/train/evaluate/predict
  4. dataset_dir: "/paddle/dataset/paddlex/table_rec/table_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: 10
  17. batch_size: 48
  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_PP-StructrureV2_SLANet_trained.pdparams
  29. Predict:
  30. model_dir: "output/best_accuracy"
  31. input_path: "/paddle/dataset/paddlex/table_rec/table_rec_dataset_examples/images/border_10368_GBUAFQNHRKR5FUQ6ZE50.jpg"
  32. kernel_option:
  33. run_mode: paddle
  34. batch_size: 1