mask-rcnn_r50-mdconv-c3-c5_fpn_amp-1x_coco.py 393 B

12345678910
  1. _base_ = '../mask_rcnn/mask-rcnn_r50_fpn_1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False),
  5. stage_with_dcn=(False, True, True, True)))
  6. # MMEngine support the following two ways, users can choose
  7. # according to convenience
  8. # optim_wrapper = dict(type='AmpOptimWrapper')
  9. _base_.optim_wrapper.type = 'AmpOptimWrapper'