当我运行keras官方文件“ cifar10_resnet.py”时,如何修复fit_generator中的“ ValueError”?

时间:2019-01-07 03:53:50

标签: keras resnet

我正在使用kears的示例"cifar10_resnet.py"运行resnet模型,但是会引发值错误。

我的Spyder版本:3.3.2 Keras版本:2.2.4 Tensorflow版本:1.11.0

model.fit_generator(datagen.flow(x_train, y_train, batch_size=batch_size),
                        validation_data=(x_test, y_test),
                        epochs=epochs, verbose=1, workers=4,
                        callbacks=callbacks)

参数datagen.flow的特定数据来自:https://github.com/keras-team/keras/blob/master/examples/cifar10_resnet.py 376-416行。

  

ValueError:steps_per_epoch=None仅对基于生成器的生成器有效   在keras.utils.Sequence类上。请指定steps_per_epoch   或使用keras.utils.Sequence类。

我猜我的设置有问题,但是我不知道如何查找和修复它。

0 个答案:

没有答案