Change dataset member variable once per epoch

I don’t know how your private DataLoader works, but this post might be helpful.