这里看你的代码是否出现了out+=y这样的代码,如:
def forward(self, x):
y = x
out = torch.nn.functional.relu(self.norm1(self.conv1(x)))
out = torch.nn.functional.relu(self.norm2(self.conv2(out)))
out = torch.nn.functional.relu(self.norm3(self.conv3(out)))
if self.bottleneck_size * 4 != self.num_channels_in:
y = self.norm4(self.conv4(x))
out += y
return torch.nn.functional.relu(out)
解决方法:将+=改成out = out+y