upd

YieldsLabs · Oct 4, 2024 · d174507 · d174507
1 parent b3e474f
commit d174507
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 356 deletions.
diff --git a/notebooks/models/autoencoder_v1.py b/notebooks/models/autoencoder_v1.py
@@ -119,7 +119,7 @@ def forward(self, x):
             latent = attn_encoded.squeeze(1)
 
         decoded = self.fc_decoder(latent)
-        decoded = decoded.view(decoded.size(0), 128, self.encoded_length)
+        decoded = decoded.view(decoded.size(0), 128, int(decoded.size(1) // 128))
         decoded = self.decoder(decoded)
         decoded = decoded.permute(0, 2, 1)