I have trained a simple sequence to sequence model and saved the model in file. Now when I loaded the model and trying to do the testing, I am getting the following error.

RuntimeError: Assertion `THCTensor_(checkGPU)(state, 3, dst, src, indices)’ failed.

I am getting error at the following line.
embedded = self.embedding(input).view(1, 1, -1)
I am guessing the problem is not in that line but rather in saving and loading process. I am running the code in the same server, in the same GPU where I have trained the model.
Does anyone know what the error means? Any suggestion to resolve this issue?