Now,  s
(1) train = test set
(2) with data augmentation + shuffling for Train set,
(3) still no shuffling and no data augmentation for test set
(4) BN with track_running_stats=False

I will now relax test set to be different than train set although originating from the same big set of images