Update NMT_with_Transformers_Reformers_using_Trax.ipynb

[sunvod]

If you choose to work with your own dataset, the generator must be infinite to prevent the train phase from being interrupted by the StopIteration event: during the generative cycle