举例:1000张图片,每次训练10张

epoch:这1000张图片全部训练完成一次即为一个epoch;
batch_size:每次训练10张图片,即 batch_size = 10;
steps:steps = 1000/10(整除则结果为steps,未能整除则结果 + 1为steps),表示更新多少次梯度。
注:这里每隔10张图片就更新一次梯度,也就是将一个batch中的图片一张张送入模型,累加loss求平均,之后进行反向传播(loss.backward()),同时进行参数更新(Iteration)。

Logo

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐