I’m not sure why you are using the model components explicitly than using the model definition file’s forward method. Any good reason for that?
What about bias terms in conv,  linear layers? Are you using them too?