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