Solved by ptrblck in post #5 
              
              
                Yes, you could use either this workflow (or the Dataset directly as mentioned by @omarfoq). 
Note, that calculating the dataset stats would only be needed once (if you have a fixed training, validation, and test split) and you could thus reuse the stats in all future experiments (as is done with th…