fovea_r50_fpn_gn-head-align_4xb4-2x_coco.py 572 B

1234567891011121314151617181920
  1. _base_ = './fovea_r50_fpn_4xb4-1x_coco.py'
  2. model = dict(
  3. bbox_head=dict(
  4. with_deform=True,
  5. norm_cfg=dict(type='GN', num_groups=32, requires_grad=True)))
  6. # learning policy
  7. max_epochs = 24
  8. param_scheduler = [
  9. dict(
  10. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  11. dict(
  12. type='MultiStepLR',
  13. begin=0,
  14. end=max_epochs,
  15. by_epoch=True,
  16. milestones=[16, 22],
  17. gamma=0.1)
  18. ]
  19. train_cfg = dict(max_epochs=max_epochs)
  20. optim_wrapper = dict(clip_grad=dict(max_norm=35, norm_type=2))