Hey @Siva_Karthikeya!
It would help if we could reduce your code down to something more manageable that still produces the discrepancy you’re observing
. In the mean time, I recommend checking out our documentation for TorchLayer (see Usage Details): qml.qnn.TorchLayer — PennyLane 0.37.0 documentation
Under “Usage Details”, there’s a section about model saving. Let me know if that helps!