Commit a8dbd3a6 authored by Mark Daoust's avatar Mark Daoust Committed by TensorFlower Gardener
Browse files

Do not override `steps_per_epoch` if it is not None. 

Currently if you pass a tf.keras.preprocessing.image.ImageDataGenerator to fit,
with a `steps_per_epoch`, your manual `steps_per_epoch` is overridden with the real value. Fix it to only override the value if the `steps_per_epoch` was not manually set.

PiperOrigin-RevId: 229774626
parent 45290705
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment