Jupyter Notebook中模型训练时epoch下方显示的数字含义是什么?
训练日志中步数字段的含义
你提到的1563/1563是深度学习训练框架的通用输出标识,和你使用的CNN模型类型无关,具体指代当前Epoch下已完成的训练步数 / 当前Epoch的总训练步数。
相关规则说明:
- 单步(step)对应模型完成1个批次(batch)数据的前向推理、损失计算、反向传播、参数更新全流程
- 单Epoch总步数的计算公式为:
训练集总样本数 ÷ 设定的batch_size,如果计算结果不是整数则向上取整
你给出的日志数值对应经典的CIFAR-10数据集训练场景:CIFAR-10训练集共50000个样本,当设置batch_size=32时,50000÷32=1562.5,向上取整后总步数就是1563,和日志输出完全匹配。
日志中同行的其他字段也可以对应理解:29s是当前Epoch总耗时,18ms/step是单步平均耗时,后续的loss、accuracy分别是当前训练集的损失值和准确率。
内容的提问来源于stack exchange,提问作者Manu Prakash Choudhary
相关产品推荐
相关产品推荐

