结合算子选择的卷积神经网络显存优化算法

  • 打印
  • 收藏
收藏成功
微博 QQ空间 微信


打开文本图片集

摘要： 针对卷积神经网络训练中自动算子选择算法在较大的显存压力下性能下降的问题， 将卸载、 重计算与卷积算子选择统一建模， 提出一种智能算子选择算法。该算法权衡卸载和重计算引入的时间开销与更快的卷积算子节省的时间， 寻找卸载、 重计算和卷积算子选择的调度， 解决了自动算子选择算法性能下降的问题. 实（剩余13375字）

试读结束

目录
monitor