feat: measuring code + graph generator code

2025-12-15 22:53:32 +01:00 · 2025-12-15 22:53:32 +01:00 · f3b07c1df3
commit f3b07c1df3
parent dd0b3d3945
6 changed files with 325 additions and 140 deletions
--- a/src/models/autoencoder/autoencoder.py
+++ b/src/models/autoencoder/autoencoder.py
@ -58,8 +58,6 @@ class AutoEncoder(Model):
        """
        x: torch.Tensor of floats
        """
-        if len(x.shape) == 2:
-            x = x.unsqueeze(1)
        return self.decoder(x)

    def forward(self, x: torch.LongTensor) -> torch.Tensor: