Hello there,
Do you have any components in your model that change when the model is put into train() or eval() mode? Do you use any custom code to save and load weights? Different settings when you reload the weights? Also, got a link to the repo?