problem when training!

[karma08021206]

thanks for the source, and I`m new in deep learning
when training the model, I got an error that
"Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same"
which means that the model was not in gpu
but when I add "model.to('cuda')" the transfer it to my gpu, I stack with another error says
"Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!"
how can I solve it?
I run the code in centos7 with nvidia 2080ti