experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #37603 from teobouvard:master

Browse Source 
PiperOrigin-RevId: 304632771
Change-Id: I59217f1fff84b6cdeda90bf8260662ec695fde5e
This commit is contained in:
TensorFlower Gardener 2020-04-03 09:41:12 -07:00
commit 569f3f82f6
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/keras/preprocessing/image.py
View File

@ -452,8 +452,8 @@ class ImageDataGenerator(image.ImageDataGenerator):
# (std, mean, and principal components if ZCA whitening is applied) # (std, mean, and principal components if ZCA whitening is applied)
datagen.fit(x_train) datagen.fit(x_train)
# fits the model on batches with real-time data augmentation: # fits the model on batches with real-time data augmentation:
model.fit_generator(datagen.flow(x_train, y_train, batch_size=32), model.fit(datagen.flow(x_train, y_train, batch_size=32),
steps_per_epoch=len(x_train) / 32, epochs=epochs) steps_per_epoch=len(x_train) / 32, epochs=epochs)
# here's a more "manual" example # here's a more "manual" example
for e in range(epochs): for e in range(epochs):
print('Epoch', e) print('Epoch', e)
@ -486,7 +486,7 @@ class ImageDataGenerator(image.ImageDataGenerator):
target_size=(150, 150), target_size=(150, 150),
batch_size=32, batch_size=32,
class_mode='binary') class_mode='binary')
model.fit_generator( model.fit(
train_generator, train_generator,
steps_per_epoch=2000, steps_per_epoch=2000,
epochs=50, epochs=50,