ValueError: steps_per_epoch=None is only valid for a generator based on the keras.utils.Sequence

最新推荐文章于 2024-09-11 21:58:36 发布

weixin_41432433

最新推荐文章于 2024-09-11 21:58:36 发布

阅读量4.4k

点赞数 1

本文链接：https://blog.csdn.net/weixin_41432433/article/details/88667571

版权

https://github.com/keras-team/keras/tree/master/examples

运行如上代码cifar10_resnet.py文件出现下列错误

ValueError: steps_per_epoch=None is only valid for a generator based on the keras.utils.Sequence class. Please specify steps_per_epoch or use the keras.utils.Sequence class.

只需要把

  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)

换成

model.fit_generator(datagen.flow(x_train, y_train, batch_size=batch_size),
                        steps_per_epoch=x_train.shape[0] // batch_size,
                        validation_data=(x_test, y_test),
                        epochs=epochs, verbose=1, workers=4,
                        callbacks=callbacks)