Thanks Thomas. I also thought of logging the mean and std when I train my model and provide them as meta data to the user who is going to use the trained model for his/her test set, so s/he can use the transofrm to normalzie before feeding them to the pre trained model.