Commit 9b55ee11 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Make `steps` and `steps_per_epoch` optional for TPUStrategy when runs_per_step is 1 

Note:
If the user passes a dataset which is repeated indefinitely and has unknown cardinality, and does not specify `steps`/`steps_per_epoch` parameter, then predict()/fit()/evaluate() will run forever.
PiperOrigin-RevId: 233135393
parent e9330cec
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