Thanks @ptrblck for your reply. Yes the forward is working correctly. But I don’t understand that in the google collab it is getting initialised the right way while in my gpu it is getting reversed. How can that be possible as I am using importing the model the same way.