should remove compute output shape of basic layer

innat · Mar 21, 2024 · eec9c41 · eec9c41
1 parent bb4cb13
commit eec9c41
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/videoswin/blocks/basic.py b/videoswin/blocks/basic.py
@@ -114,24 +114,24 @@ 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 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)