Solved by ptrblck in post #2 
              
              
                I think the common use case would be to use it during the training only. 
The validation dataset should act as a good proxy for the final model performance on the unseen test data. So I would try to keep the data distribution of the validation and test datasets as close as possible. 
Since your test…