Fixed the bug validation_freq is not passed to TPUStrategy loop.
PiperOrigin-RevId: 233252849
This commit is contained in:
parent
43947b9efc
commit
8658f093ce
@ -113,7 +113,7 @@ def fit_distributed(model,
|
||||
initial_epoch=initial_epoch,
|
||||
steps_per_epoch=steps_per_epoch,
|
||||
validation_steps=validation_steps,
|
||||
validation_freq=1)
|
||||
validation_freq=validation_freq)
|
||||
else:
|
||||
return training_arrays.fit_loop(
|
||||
model,
|
||||
|
Loading…
Reference in New Issue
Block a user