if you want to do that you need to define two ImageDataGEnerator objects one for train and one for validation set and use them as per your need.