mmdetection使用混合精度训练

在配置文件中加入如下内容即可:

 # 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

https://zhuanlan.zhihu.com/p/375224982

你可能感兴趣的:(深度学习,人工智能)