Say I have trained a model with some normalized train data. When inference, I need the same mean and std of the normalization. Naturally, this mean and std are stored in my data module which is not included in the checkpoint.
Now I have to put the mean/std in my model as 2 parameters of ini in which I call self.save_hyparameters() to save them.
Any suggestions?