Yeah. thanks ! I thought the same. I have been talking to some people and the errors might be because the GPU I am trying to use is deprecated so I will set the whole thing up in a new server/GPU. Thanks for replies
EDIT: The error was in the observed data, I forgot to .cuda() it (In can help someone)