Commit 4225a3ad authored by Rohan Jain's avatar Rohan Jain Committed by TensorFlower Gardener
Browse files

Resetting the MultiDeviceIterator in eager mode instead of creating a new one... 

Resetting the MultiDeviceIterator in eager mode instead of creating a new one at the end of each epoch. This saves the memory used in creating new functions each time we create a new MultiDeviceIterator.

PiperOrigin-RevId: 235071336
parent ba90c2b6
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