2020年8月9日
Pytorch原生AMP支持使用方法(1.6版本)
摘要： AMP：Automatic mixed precision，自动混合精度，可以在神经网络推理过程中，针对不同的层，采用不同的数据精度进行计算，从而实现节省显存和加快速度的目的。 在Pytorch 1.5版本及以前，通过NVIDIA出品的插件apex，可以实现amp功能。 从Pytorch 1.6版本 阅读全文
posted @ 2020-08-09 22:49 LyleChen 阅读(4974) 评论(0) 推荐(2) 编辑