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(
|
return model.fit(
|
||||||
x=train_ds,
|
x=train_ds,
|
||||||
epochs=hparams.train_epochs,
|
epochs=hparams.train_epochs,
|
||||||
steps_per_epoch=hparams.steps_per_epoch,
|
|
||||||
validation_data=validation_ds,
|
validation_data=validation_ds,
|
||||||
callbacks=callbacks)
|
callbacks=callbacks)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user