ppyolo_r50vd_dcn_1x_minicoco.yml 849 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. './_base_/ppyolo_r50vd_dcn.yml',
  5. './_base_/optimizer_1x.yml',
  6. './_base_/ppyolo_reader.yml',
  7. ]
  8. snapshot_epoch: 8
  9. use_ema: true
  10. weights: output/ppyolo_r50vd_dcn_1x_minicoco/model_final
  11. TrainReader:
  12. batch_size: 12
  13. TrainDataset:
  14. !COCODataSet
  15. image_dir: train2017
  16. # refer to https://github.com/giddyyupp/coco-minitrain
  17. anno_path: annotations/instances_minitrain2017.json
  18. dataset_dir: dataset/coco
  19. data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
  20. epoch: 192
  21. LearningRate:
  22. base_lr: 0.005
  23. schedulers:
  24. - !PiecewiseDecay
  25. gamma: 0.1
  26. milestones:
  27. - 153
  28. - 173
  29. - !LinearWarmup
  30. start_factor: 0.
  31. steps: 4000
  32. OptimizerBuilder:
  33. optimizer:
  34. momentum: 0.9
  35. type: Momentum
  36. regularizer:
  37. factor: 0.0005
  38. type: L2