深度学习中epoch、batch_size、steps的含义

剑哥的后仰跳投

2002人浏览 · 2022-12-06 16:25:11

剑哥的后仰跳投 · 2022-12-06 16:25:11 发布

举例：1000张图片，每次训练10张

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

魔乐社区

魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。

更多推荐

小参数・大码力・易部署 | Qwen3.6-27B上线魔乐社区，基于昇腾的部署教程来了

继一周前模型开源发布后，千问再度开源Qwen3.6-27B —— 一个拥有270亿参数的稠密多模态模型，也是社区呼声最高的模型规格。Qwen3.6-27B 依然支持多模态思考与非思考模式，在智能体编程方面达到了旗舰级表现，全面超越前代开源旗舰 Qwen3.5-397B-A17B（总参数397B / 激活参数17B的MoE模型）。作为稠密架构，它无需MoE路由即可部署，是开发者在实用、可广泛部署规模