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

Fix end-of-epoch tf.data.Datasets reset code. 

The datasets weren't getting reset, because the loop was was reading `output_generator`,
but the reset path was only resetting `generator`.

PiperOrigin-RevId: 232564221
parent 5943cddd
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