RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the

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):