Thanks for replying @iffiX
I will try that, but on a side note, I have passed model.parameter() to the optimizer that technically should update the parameters right irrespective of self.forward() or self.call … am I right ?