SLANeXt_wireless.yaml 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. Global:
  2. model: SLANeXt_wireless
  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: 16
  18. learning_rate: 0.001
  19. pretrain_weight_path: "https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/SLANeXt_wireless_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/SLANeXt_wireless_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/table_recognition.jpg"
  33. kernel_option:
  34. run_mode: paddle