batch_size = 32 为什么运行后每一个 Epoch 下不是显示 32 个批次，而只是显示一个批次？

Question

老师，下午好！
RT,

history = model.fit(x_train, y_train, epochs = 10, batch_size = 32,
                    validation_data = (x_valid, y_valid))

输出结果是：

Epoch 1/10
1719/1719 [==============================] - 3s 2ms/step - loss: 13.4092 - accuracy: 0.6626 - val_loss: 0.5988 - val_accuracy: 0.7938
Epoch 2/10

为什么会是上面这样，
根据 https://mp.weixin.qq.com/s/FFF6RSudAas7j2vHqP5j2Q
输出结果应该是：

Epoch 1/10
1719/1719 [==============================] - 3s 2ms/step - loss: 13.4092 - accuracy: 0.6626 - val_loss: 0.5988 - val_accuracy: 0.7938
1719/1719 [==============================] - 3s 2ms/step - loss: 13.4092 - accuracy: 0.6626 - val_loss: 0.5988 - val_accuracy: 0.7938
1719/1719 [==============================] - 3s 2ms/step - loss: 13.4092 - accuracy: 0.6626 - val_loss: 0.5988 - val_accuracy: 0.7938
...
这样一共有32次才到
Epoch 2/10 
呀？

谢谢老师解答！！！

Answer 1

正十七 2021-04-14 11:30:37

先别管那篇文章里怎么说的，在tf里，

一个epoch指的是遍历整个数据集一遍。

batch_size是每一步需要多少个样本。

这样，一个epoch需要的步数就是 总样本数 / batch_size = 55000 / 32 = 1719。

所以走完1719步后，一个epoch就完成了。

• 提问者 慕神4535282 #1

  非常感谢！