PicoDet_layout_1x.yaml 630 B

12345678910111213141516171819202122232425262728293031
  1. Global:
  2. model: PicoDet_layout_1x
  3. mode: check_dataset # check_dataset/train/evaluate
  4. dataset_dir: "/paddle/dataset/paddlex/layout/det_layout_examples"
  5. device: gpu:0,1,2,3
  6. output: "output"
  7. CheckDataset:
  8. dst_dataset_name: dataset
  9. convert:
  10. enable: False
  11. src_dataset_type: nul
  12. split:
  13. enable: False
  14. train_percent: null
  15. val_percent: null
  16. Train:
  17. num_classes: 4
  18. epochs_iters: 50
  19. batch_size: 24
  20. learning_rate: 0.4
  21. pretrain_weight_path: null
  22. warmup_steps: 100
  23. resume_path: null
  24. log_interval: 1
  25. eval_interval: 1
  26. Evaluate:
  27. weight_path: "output/best_model.pdparams"
  28. log_interval: 10