diff --git a/peleenet.py b/peleenet.py index ec92edc..ebef7ab 100644 --- a/peleenet.py +++ b/peleenet.py @@ -72,7 +72,7 @@ def forward(self, x): out2 = self.branch2a(x) out2 = self.branch2b(out2) - out3 = self.branch2c(out2) + out2 = self.branch2c(out2) out = torch.cat([x, out1, out2], dim=1) return out