我下面有代码，
train_datagen = ImageDataGenerator(
rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
test_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
'data/train',
target_size=(150, 150),
batch_size=32,
class_mode='binary')
validation_generator = test_datagen.flow_from_directory(
'data/validation',
target_size=(150, 150),
batch_size=32,
class_mode='binary')
现在model.fit_generator定义如下：
model.fit_generator(
train_generator,
steps_per_epoch=2000,
epochs=50,
validation_data=validation_generator,
validation_steps=800)
现在model.fit_generator已弃用，在这种情况下更改model.fit_generator为正确的方法是什么model.fit？
您只需更改model.fit_generator()为model.fit()。
从TensorFlow 2.1开始，model.fit()还接受生成器作为输入。就如此容易。
来自TensorFlow的官方文档：
警告：此功能已被弃用。它将在将来的版本中删除。更新说明：请使用支持生成器的Model.fit。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句