fovea_r101_fpn_gn-head-align_4xb4-2x_coco.py 650 B

1234567891011121314151617181920212223
  1. _base_ = './fovea_r50_fpn_4xb4-1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. depth=101,
  5. init_cfg=dict(type='Pretrained',
  6. checkpoint='torchvision://resnet101')),
  7. bbox_head=dict(
  8. with_deform=True,
  9. norm_cfg=dict(type='GN', num_groups=32, requires_grad=True)))
  10. # learning policy
  11. max_epochs = 24
  12. param_scheduler = [
  13. dict(
  14. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  15. dict(
  16. type='MultiStepLR',
  17. begin=0,
  18. end=max_epochs,
  19. by_epoch=True,
  20. milestones=[16, 22],
  21. gamma=0.1)
  22. ]
  23. train_cfg = dict(max_epochs=max_epochs)