将ImageDataGenerator与model.fit_generator()一起使用

时间:2017-06-25 23:33:20

标签: keras

我目前正在使用此代码进行培训:

model.fit_generator()

如何将其转换为使用{{1}}并采用相同的方式?

1 个答案:

答案 0 :(得分:0)

gen = generator.flow(x_train, y_train, batch_size=10240)

model.fit_generator(gen, steps_per_epoch=number_of_samples//batch_size, ...)