retinanet_r50-caffe_fpn_ms-2x_coco.py 388 B

12345678910111213141516
  1. _base_ = './retinanet_r50-caffe_fpn_ms-1x_coco.py'
  2. # training schedule for 2x
  3. train_cfg = dict(max_epochs=24)
  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=24,
  12. by_epoch=True,
  13. milestones=[16, 22],
  14. gamma=0.1)
  15. ]