grid-rcnn_r50_fpn_gn-head_1x_coco.py 414 B

12345678910111213141516171819
  1. _base_ = './grid-rcnn_r50_fpn_gn-head_2x_coco.py'
  2. # training schedule
  3. max_epochs = 12
  4. train_cfg = dict(max_epochs=max_epochs)
  5. # learning rate
  6. param_scheduler = [
  7. dict(
  8. type='LinearLR', start_factor=0.0001, by_epoch=False, begin=0,
  9. end=500),
  10. dict(
  11. type='MultiStepLR',
  12. begin=0,
  13. end=max_epochs,
  14. by_epoch=True,
  15. milestones=[8, 11],
  16. gamma=0.1)
  17. ]