fix errors

jackyin68 · Jul 15, 2019 · e97d6b5 · e97d6b5
1 parent e41fa56
commit e97d6b5
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/models/resnet_models.py b/models/resnet_models.py
@@ -237,12 +237,8 @@ def forward(self, x):
         x = self.layer3(x)
         x = self.layer4(x)
 
-        if self.isRefine:
-            res1x, res2x, res4x = self.layer5(x, input_size)
-            return res1x, res2x, res4x
-        else:
-            res = self.layer5(x, input_size)
-            return res
+        res = self.layer5(x, input_size)
+        return res
 
     def train(self, mode=True, freezeBn=True):
         super(ResNet, self).train(mode=mode)