faster-rcnn_r50_fpn_amp-1x_coco.py 225 B

123456
  1. _base_ = './faster-rcnn_r50_fpn_1x_coco.py'
  2. # MMEngine support the following two ways, users can choose
  3. # according to convenience
  4. # optim_wrapper = dict(type='AmpOptimWrapper')
  5. _base_.optim_wrapper.type = 'AmpOptimWrapper'