SLANet_plus.yaml 1.0 KB

1234567891011121314151617181920212223242526272829303132
  1. Global:
  2. model: SLANet_plus
  3. mode: check_dataset # check_dataset/train/evaluate/predict
  4. dataset_dir: "/Users/zhch158/workspace/repository.git/PaddleX/dataset/table_rec_dataset_examples"
  5. device: cpu
  6. output: "/Users/zhch158/workspace/repository.git/PaddleX/zhch/table_structure_recognition/output"
  7. use_sync_bn: False # 移动到这里!
  8. distributed: False # 也添加这个确保不使用分布式
  9. CheckDataset:
  10. convert:
  11. enable: False
  12. src_dataset_type: null
  13. split:
  14. enable: False
  15. train_percent: null
  16. val_percent: null
  17. Train:
  18. epochs_iters: 10
  19. batch_size: 8 # 减少batch_size,CPU训练建议使用较小值
  20. learning_rate: 0.001
  21. pretrain_weight_path: https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SLANet_plus_pretrained.pdparams
  22. resume_path: null
  23. log_interval: 20
  24. eval_interval: 1
  25. save_interval: 1
  26. Evaluate:
  27. # weight_path: "output/best_accuracy/best_accuracy.pdparams"
  28. weight_path: "/Users/zhch158/.paddleocr/models/SLANet_plus_pretrained.pdparams"
  29. log_interval: 1