From b306ca7ee1a8e1d4c4ad8f3a081b6e81c2c8f37d Mon Sep 17 00:00:00 2001
From: Joshua Batson <joshua.batson@czbiohub.org>
Date: Mon, 16 Dec 2019 20:31:23 -0800
Subject: [PATCH] Fix missing application of conv2 in babyunet

---
 models/babyunet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/babyunet.py b/models/babyunet.py
index 6115bd9..f2f5b9a 100644
--- 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)