mask-rcnn_x50-32x4d_fpn_gn-ws-all_2x_coco.py 559 B

12345678910111213141516171819
  1. _base_ = './mask-rcnn_r50_fpn_gn-ws-all_2x_coco.py'
  2. # model settings
  3. conv_cfg = dict(type='ConvWS')
  4. norm_cfg = dict(type='GN', num_groups=32, requires_grad=True)
  5. model = dict(
  6. backbone=dict(
  7. type='ResNeXt',
  8. depth=50,
  9. groups=32,
  10. base_width=4,
  11. num_stages=4,
  12. out_indices=(0, 1, 2, 3),
  13. frozen_stages=1,
  14. style='pytorch',
  15. conv_cfg=conv_cfg,
  16. norm_cfg=norm_cfg,
  17. init_cfg=dict(
  18. type='Pretrained',
  19. checkpoint='open-mmlab://jhu/resnext50_32x4d_gn_ws')))