retinanet_r50-caffe_fpn_ms-3x_coco.py 389 B

1234567891011121314151617
  1. _base_ = './retinanet_r50-caffe_fpn_ms-1x_coco.py'
  2. # training schedule for 2x
  3. train_cfg = dict(max_epochs=36)
  4. # learning rate policy
  5. param_scheduler = [
  6. dict(
  7. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  8. dict(
  9. type='MultiStepLR',
  10. begin=0,
  11. end=36,
  12. by_epoch=True,
  13. milestones=[28, 34],
  14. gamma=0.1)
  15. ]