Set steps_per_epoch
to None when calling model.fit() method for image classifier.
PiperOrigin-RevId: 483764377
This commit is contained in:
parent
36bd9abb8f
commit
a28c9d2c26
|
@ -98,6 +98,5 @@ def train_model(model: tf.keras.Model, hparams: hp.HParams,
|
|||
return model.fit(
|
||||
x=train_ds,
|
||||
epochs=hparams.train_epochs,
|
||||
steps_per_epoch=hparams.steps_per_epoch,
|
||||
validation_data=validation_ds,
|
||||
callbacks=callbacks)
|
||||
|
|
Loading…
Reference in New Issue
Block a user