RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same. Using CNN for the project

Question

I am trying to run my CNN model but keep getting the same error with cuda(), and not sure on how to fix it by myself... the code below is the code I use. I am using the dataset