mask-rcnn_r101_fpn_gn-all_3x_coco.py 404 B

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