go back to 256x256 inputs

Former-commit-id: 0a8cbd9cf9d2a80e143b5c41d6e33e029fec53ed
javiribera · Feb 25, 2018 · e4a65a4 · e4a65a4
1 parent f2d0f8f
commit e4a65a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/unet_model.py b/unet_model.py
@@ -32,7 +32,7 @@ def __init__(self, n_channels, n_classes,
 
         self.known_n_points = known_n_points
         if known_n_points is None:
-            self.regressor = nn.Linear(288*384, 1)
+            self.regressor = nn.Linear(256*256, 1)
             self.regressor_nonlin = nn.Softplus()
 
     def forward(self, x):