Closed sunvod closed 2 years ago
Thanks for your pull request. It looks like this may be your first contribution to a Google open source project (if not, look below for help). Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).
:memo: Please visit https://cla.developers.google.com/ to sign.
Once you've signed (or fixed any issues), please reply here with @googlebot I signed it!
and we'll verify it.
ℹ️ Googlers: Go here for more info.
@googlebot I signed it!
With the solution posted previously, if the training consumes all the inputs provided by the generator, a StopIteration is launched and the execution ends. The proposed solution avoids stopping execution without necessarily loading all the data into memory.