Nov, 2018

深度学习模型计算成本的预测

TL;DR本文提出了一种针对深度学习神经网络训练时间预测的方法,通过训练神经网络预测每部分的执行时间,从而得出整体的深度学习网络训练时间,可以用于推断每个批次或整个epoch的执行时间,也可以支持选择适当的硬件和模型。