张量流警告：您的输入数据不足

Question

晚上好，在第一个纪元即将结束时，我一直在 tensorflow 上收到这个警告。 '警告：tensorflow：你的输入数据用完了； 中断训练。 确保您的数据集或生成器至少可以生成steps_per_epoch * epochs批次（在本例中为 78 个批次）。 在构建数据集时，您可能需要使用 repeat() 函数。

下面我报告我的模型。

 INIT_LR = 0.0001 EPOCHS = 10 BS=32 aug = ImageDataGenerator( rotation_range=45, width_shift_range=0.2, height_shift_range=0.2, rescale=1./255, horizontal_flip=True, fill_mode="nearest") headModel = baseModel.output headModel = AveragePooling2D(pool_size=(7, 7))(headModel) headModel = Flatten(name="flatten")(headModel) headModel = Dense(128, activation="relu")(headModel) headModel = Dropout(0.5)(headModel) headModel = Dense(2, activation="softmax")(headModel) model = Model(inputs=baseModel.input, outputs=headModel) for layer in baseModel.layers: layer.trainable = False print("[INFO] compiling model...") opt = Adam(lr=INIT_LR) model.compile(loss="binary_crossentropy", optimizer=opt, metrics=["accuracy"]) print("[INFO] training head...") H = model.fit( aug.flow(trainX, trainY, batch_size=BS), steps_per_epoch=len(trainX) // BS, validation_data=(testX, testY), validation_steps=len(testX) // BS, epochs=EPOCHS)

我还是不明白为什么。 我已经把steps_per_epoch = steps_per_epoch=len(trainX) // BS 和validation_steps 类似，但是这个警告仍然存在，你知道可能是什么问题吗？ 我正在使用 10.000 个样本的数据集，数据分为 75% 训练和 25% 有效

Answer 1

警告不会损害您的代码执行。 您可以使用以下代码抑制这些警告

import logging, os
logging.disable(logging.WARNING)
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
import tensorflow as tf