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

it2026-02-18  5

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

model.cuda()

或者:

model.to('cuda')

 

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

self.head_convs = [ ]

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

self.head_convs = nn.ModuleList()

最新回复(0)