Dataset size vs bactch x steps / epoch

[jdeschamps]

From @tibuch in #40: Datasets are currently limited, which leads to problems if steps_per_epoch x batch_size is larger than the dataset length. As long as steps_per_epoch x batch_size is smaller than len(dataset) everything works fine.