Shuffle dataset with larger granularity

Hi Frank,
Actually this post related to my other post Finetuning intermediate layers of resnet18. take a look at that post and that may make more sense what I’m doing.