Fix reshape shape

TimKoornstra · Oct 28, 2024 · cb40dcd · cb40dcd
1 parent cea67b3
commit cb40dcd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -13,7 +13,7 @@
 project = 'VGSLify'
 copyright = '2024, Tim Koornstra'
 author = 'Tim Koornstra'
-release = '0.13.2'
+release = '0.13.3'
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 import os
 
-version = "0.13.2"
+version = "0.13.3"
 
 
 # Function to write the version to a _version.py file

diff --git a/vgslify/core/factory.py b/vgslify/core/factory.py
@@ -436,7 +436,7 @@ def reshape(self, spec: str):
                 else:
                     seq_length = H * W if W is not None else None
 
-                features = C
+                features = W * C
                 config = ReshapeConfig(target_shape=(seq_length, features))
                 layer = self._reshape(config)
                 self.layers.append(layer)