rollback

MannLabs · May 16, 2024 · 1fc84fc · 1fc84fc
1 parent 123fc95
commit 1fc84fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
@@ -201,7 +201,7 @@ def forward(self, x):
 
         print("x value" , x.item())
 
-        return x.squeeze()
+        return x
 
 ### CAE Model Architecture