RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor)
I am trying to implement CUDA for this graph neural network model “https://github.com/VincLee8188/GMAN-PyTorch” to use the nvidia GPU to be able to speed up the process since with the CPU it takes about 10 hours, all this I am using kaggle and I wanted to be able to use their GPU. I started to implement it but I get this error as soon as the training model is started. I am inserting below the main, model and train. Do you have any ideas on how I can fix this error?