attention-transfer
attention-transfer copied to clipboard
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
When I ran the cifar code with GPU, I got the following error. Any suggestions would be appreciated!
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
Hi. Your model has not been transferred to GPU yet.