Pytorch出现RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)

一般来说遇到这个问题可能是没有执行：

model.cuda()

或者：

model.to('cuda')

但是我这次遇到的并不是因为以上原因，而是因为:

self.head_convs = [ ]

这里应该用 nn.ModuleList 而不是普通的 python list。使用后者会导致 self.head_convs 中存放的nn.Module 无法进入GPU。因此，应该改成如下：

self.head_convs = nn.ModuleList()