简介

在迁移完成、准备训练之前,需要开启混合精度,用户可以根据场景选择引入APEX混合精度模块或使用PyTorch1.8.1框架自带的AMP功能,以保证模型的性能。APEX模块有4种功能模式可选择,可以提供不同场景下的混合精度训练支持,而AMP功能只类似APEX模块中的一种功能,但无需引入可以直接使用。