《表2 模型训练时批归一化计算量[2]、访存及计算密度》
模型训练时,批归一化层的输入与输出数据量相同,假设前向计算过程中批归一化层的输入数据个数为E×F×N且已经保存于主存中(前向的卷积结果),其计算使用单精度(4B)[9]。依据上述分析,统计训练时批归一化层的计算量、访存量及计算密度,如表2所示。
图表编号 | XD00139896800 严禁用于非法目的 |
---|---|
绘制时间 | 2020.04.20 |
作者 | 王吉军、郝子宇、李宏亮 |
绘制单位 | 江南计算技术研究所、江南计算技术研究所、江南计算技术研究所 |
更多格式 | 高清、无水印(增值服务) |