Could you compare the model outputs using a static input (e.g. torch.ones) before saving the model and after loading it? If these two outputs are equal up to the limited numerical precision of floating point numbers, I would guess that the data preprocessing might be different.
PS: you can post code snippets by wrapping them into three backticks ```