mmdetection/configs/mask_rcnn/mask-rcnn_r50_fpn_amp-1x_coco.py

_base_ = './mask-rcnn_r50_fpn_1x_coco.py'

# Enable automatic-mixed-precision training with AmpOptimWrapper.
optim_wrapper = dict(type='AmpOptimWrapper')