求助，跑tensorflow时样本莫名少了。

Question

用tensorflow跑fashion mnist理应有60000个样本，但是电脑只跑了1875个，不知道是什么问题。

 

我是照搬教程原样代码的。

这是教程的结果：

这是我的结果：

这是代码

import tensorflow as tf
import numpy as np
from tensorflow import keras
print(f"It's tensorflow {tf.__version__}")
mnist=keras.datasets.fashion_mnist
(training_images,training_labels),(test_images,test_labels)=mnist.load_data()
training_images=training_images/255
test_images=test_images/255
model=keras.Sequential()
model.add(keras.layers.Flatten())
model.add(keras.layers.Dense(128,activation='relu'))
model.add(keras.layers.Dense(10,activation='softmax'))
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy')
model.fit(training_images,training_labels,epochs=5)

Answer 1

1875不是样本个数，而是steps。

steps跟batch_size有关，model.fit中的参数model.fit默认值是32，所以steps=60000 / 32 = 1875。

可以自己设置batch_size，例如设置batch_size=1，则steps=60000。

model.fit(training_images,training_labels,epochs=5,batch_size=1)