Better shuffling
Minor fix, it seems batches are shuffled only once at the beginning of training, this modifies the batch shuffling to be done each epoch.
Minor fix, it seems batches are shuffled only once at the beginning of training, this modifies the batch shuffling to be done each epoch.