conv layer dim

MannLabs · May 17, 2024 · 45ee884 · 45ee884
1 parent 51fabaf
commit 45ee884
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
@@ -70,10 +70,7 @@ def make_layers_MLP(self, cfg_MLP, cfg, regression = False):
             nn.Sequential: A sequential model representing the MLP architecture.
         """
         # get output feature size of CNN with chosen configuration
-        if regression: 
-            in_features = int(cfg[-2]) * 4 * 4 
-        else:
-            in_features = int(cfg[-2]) * 2 * 2
+        in_features = int(cfg[-2]) * 2 * 2
 
         layers = []
         i = 0

diff --git a/src/sparcscore/ml/plmodels.py b/src/sparcscore/ml/plmodels.py
@@ -220,8 +220,6 @@ def training_step(self, batch):
     def validation_step(self, batch):
         data, target = batch
         target = target.unsqueeze(1)
-        print("target val ", target)
-
         output = self.network(data)
         loss = F.mse_loss(output, target)