Fix missing application of conv2 in babyunet

czbiohub-sf · Dec 17, 2019 · b306ca7 · b306ca7
1 parent 2e351cb
commit b306ca7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/babyunet.py b/models/babyunet.py
@@ -29,7 +29,7 @@ def forward(self, x):
         x = self.pool1(c1)
         c2 = self.conv2(x)
         x = self.pool2(c2)
-        self.conv3(x)
+        x = self.conv3(x)
 
         x = self.up1(x)
         x = torch.cat([x, c2], 1)