🎨 can it take non square input

videoswin/backbone.py

@@ -98,12 +98,12 @@ def __init__(
                 "Depth, height and width of the video must be specified"
                 " in `input_shape`."
             )
-        if input_spec.shape[-3] != input_spec.shape[-2]:
-            raise ValueError(
-                "Input video must be square i.e. the height must"
-                " be equal to the width in the `input_shape`"
-                " tuple/tensor."
-            )
+        # if input_spec.shape[-3] != input_spec.shape[-2]:
+        #     raise ValueError(
+        #         "Input video must be square i.e. the height must"
+        #         " be equal to the width in the `input_shape`"
+        #         " tuple/tensor."
+        #     )
 
         x = input_spec
 

videoswin/blocks/basic.py

@@ -114,25 +114,6 @@ def build(self, input_shape):
 
         self.built = True
 
-    # def compute_output_shape(self, input_shape):
-    #     if self.downsample is not None:
-    #         # TODO: remove tensorflow dependencies.
-    #         # GitHub issue: https://github.com/keras-team/keras/issues/19259 # noqa: E501
-    #         # output_shape = tf.TensorShape(
-    #         #     [
-    #         #         input_shape[0],
-    #         #         self.depth_pad,
-    #         #         self.height_pad // 2,
-    #         #         self.width_pad // 2,
-    #         #         2 * self.input_dim,
-    #         #     ]
-    #         # )
-
-    #         output_shape = (input_shape[0], ) + (self.depth_pad, ) + (self.height_pad // 2, ) + (self.width_pad // 2, ) + (self.input_dim, )
-    #         return output_shape
-
-    #     return input_shape
-
     def call(self, x, training=None):
         input_shape = ops.shape(x)
         batch_size, depth, height, width, channel = (