remove flatten

MannLabs · May 17, 2024 · dd23edf · dd23edf
1 parent 56260ac
commit dd23edf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
@@ -191,7 +191,7 @@ def forward(self, x):
         x = self.features(x)
         print("x.shape after features", x.shape)
 
-        x = torch.flatten(x, 1)
+        #x = torch.flatten(x, 1)
         print("x.shape after flatten", x.shape)
 
         x = self.classifier(x)