RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
- 首先检查是否数据和模型是否都是cuda挂载在GPU上了,如果不满足这个可以用网上普遍方法解决。
- 排除第一项后,检查是否nn.Module的模块是否是在初始化__init__当中创建的,如果在后续forward中更改可能会造成该问题
原代码
class model(nn.Module): # 关键代码
def __init__(self, cin):
super(Disentangler, self).__init__()
self.activation_head = nn.Conv2d(cin, 1, kernel_size=3, padding=1, bias=False)
self.bn_head = nn.BatchNorm2d(1)
def forward(self, x, inference=False):
修改后代码
class model(nn.Module): # 关键代码
def __init__(self, cin):
super(Disentangler, self).__init__()
self.activation_head = nn.Conv2d(cin, 1, kernel_size=3, padding=1, bias=False).cuda()
self.bn_head = nn.BatchNorm2d(1).cuda() # 注意要两层都改
def forward(self, x, inference=False):