Reading small datasets forces reshuffle and extra epochs despite number of epochs set

[XapaJIaMnu]

When testing with very short datasets (10 lines), the trainer wraps the dataset 10 times before it stops reading so that it can fill in a full batch.

Unfortunately this won't be as easy as the dataset itself has no notion of the maximum number of epochs it's supposed to read so it can't throw StopIteration exception