fix a bug

speedinghzl · web-flow · commit cf1d48afe9c6 · 2017-10-19T10:13:55.000+08:00
#4
diff --git a/deeplab/model.py b/deeplab/model.py
@@ -113,7 +113,7 @@ def forward(self, x):
         out = self.conv2d_list[0](x)
         for i in range(len(self.conv2d_list)-1):
             out += self.conv2d_list[i+1](x)
-            return out
+        return out
 
 class Residual_Covolution(nn.Module):
     def __init__(self, icol, ocol, num_classes):
