diff --git a/models/ResNet.py b/models/ResNet.py index ab976fb..efebc1c 100644 --- a/models/ResNet.py +++ b/models/ResNet.py @@ -31,7 +31,7 @@ def forward(self, x): x = self.bn2(x) if self.downsample is not None: - residual = self.downsample(x) + residual = self.downsample(residual) x += residual x = self.relu(x)