Hi,
You are most probably just moving the inputs to CUDA but not the model --aka the weights. It might also be that the model initialization is faulty as you have used nn.parameters inside the ‘forward’ function. Can you please edit your original question with the model architectural code and the place where you are sending over the model to CUDA ?