squeeze

MannLabs · May 16, 2024 · 1c84ee4 · 1c84ee4
1 parent 1fc84fc
commit 1c84ee4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
@@ -199,7 +199,9 @@ def forward(self, x):
 
         print(x, x.shape)
 
-        print("x value" , x.item())
+        x_sq = torch.squeeze(x)
+
+        print(x_sq, x_sq.shape)
 
         return x