Thanks for the super-fast replies. I was having my lunch break and now I'm back at my desk. `consume_epochs` specifies whether you want to get the same epoch again and again (`consume_epochs=True`), or if you want to get new epochs. But I think this has nothing to do with breaking out of the loop - I can check