From 1fc84fc63bc2b1a0b0f6edbf2c6a11294573f174 Mon Sep 17 00:00:00 2001
From: namsaraeva <a.namsaraeva@tum.de>
Date: Thu, 16 May 2024 17:07:59 +0200
Subject: [PATCH] rollback

---
 src/sparcscore/ml/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sparcscore/ml/models.py b/src/sparcscore/ml/models.py
index 569c215..cbd70ff 100644
--- 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