diff --git a/model.py b/model.py index 80d94db..b0e7c10 100644 --- a/model.py +++ b/model.py @@ -45,7 +45,7 @@ def forward(self, x): residual = self.conv1x1(x) out = self.relu(self.bn1(self.conv1(x))) out = self.relu(self.bn2(self.conv2(out))) - out += residual + out = out + residual out = self.relu(out) return out class SFConv(nn.Module): @@ -139,7 +139,7 @@ def forward(self, x): residual = self.conv1x1(x) out = self.relu(self.bn1(self.conv1(x))) out = self.relu(self.bn2(self.conv2(out))) - out += residual + out = out + residual out = self.relu(out) return out class ER_Net(nn.Module):