在配置文件中加入如下内容即可:
# fp16 settings
fp16 = dict(loss_scale=512.)
# 表示静态 scale
fp16 = dict(loss_scale=512.)
# 表示动态 scale
fp16 = dict(loss_scale='dynamic')
参考:
where is Mixed Precision Training set in config · Issue #2204 · open-mmlab/mmdetection · GitHub