add support arbitrary input shape

innat · Mar 22, 2024 · e5f443f · e5f443f
1 parent 4bf9d9b
commit e5f443f
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/videoswin/layers/patch_embed.py b/videoswin/layers/patch_embed.py
@@ -62,14 +62,6 @@ def call(self, x):
 
         return x
 
-    def compute_output_shape(self, input_shape):
-        spatial_dims = [
-            (dim - self.patch_size[i]) // self.patch_size[i] + 1
-            for i, dim in enumerate(input_shape[1:-1])
-        ]
-        output_shape = (input_shape[0],) + tuple(spatial_dims) + (self.embed_dim,)
-        return output_shape
-
     def get_config(self):
         config = super().get_config()
         config.update(