Deeplabv3_Plus-R101.yaml 639 B

12345678910111213141516171819202122232425262728293031
  1. Global:
  2. model: Deeplabv3_Plus-R101
  3. mode: check_dataset # check_dataset/train/evaluate
  4. dataset_dir: "/paddle/dataset/paddlex/seg/seg_optic_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: null
  12. split:
  13. enable: False
  14. train_percent: null
  15. val_percent: null
  16. Train:
  17. epochs_iters: 500
  18. num_classes: 2
  19. batch_size: 2
  20. learning_rate: 0.01
  21. pretrain_weight_path: null
  22. warmup_steps: 0
  23. resume_path: null
  24. log_interval: 10
  25. eval_interval: 100
  26. Evaluate:
  27. weight_path: "output/best_model/model.pdparams"
  28. log_interval: 10