TensorFlow训练MNIST时输出的1875数值来源与含义问询
日志中1875数值的来源与含义
- 数值来源
你提供的代码调用model.fit()时没有手动指定batch_size(批大小)参数,TensorFlow Keras中该参数的默认值为32。已知MNIST训练集总共有60000张样本,用总样本数除以批大小计算可得:60000 ÷ 32 = 1875,和日志中的数值完全匹配。
如果手动指定了batch_size参数,该数值也会对应变化,比如设置batch_size=64时,单轮训练的步数就会变为60000 ÷ 64 ≈ 938。 - 具体含义
该数值代表单轮(Epoch)训练的总迭代步数。每一步迭代都会处理1个包含32张图片的批次,依次完成前向推理、损失计算、反向传播更新参数的完整流程。当1875步全部跑完时,就代表60000张训练图片已经全部被模型学习过一次,当前轮次的训练完成。
内容的提问来源于stack exchange,提问作者Milind
相关产品推荐
相关产品推荐

