Mask-RT-DETR-H.yaml 618 B

12345678910111213141516171819202122232425262728293031
  1. Global:
  2. model: Mask-RT-DETR-H
  3. mode: check_dataset # check_dataset/train/evaluate
  4. dataset_dir: "/paddle/dataset/paddlex/instance_seg"
  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: 2
  18. epochs_iters: 40
  19. batch_size: 2
  20. learning_rate: 0.00005
  21. pretrain_weight_path: null
  22. warmup_steps: 100
  23. resume_path: null
  24. log_interval: 10
  25. eval_interval: 1
  26. Evaluate:
  27. weight_path: "output/best_model.pdparams"
  28. log_interval: 10