Skip to content

Commit

Permalink
rm shape info [ski ci]
Browse files Browse the repository at this point in the history
  • Loading branch information
innat committed Mar 22, 2024
1 parent 7ea8589 commit 8cce221
Showing 1 changed file with 16 additions and 16 deletions.
32 changes: 16 additions & 16 deletions videoswin/blocks/basic.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,22 +134,22 @@ def call(self, x, training=None):

return x

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,
# ]
# )
# 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,
# # ]
# # )
# # return output_shape

# return input_shape

def get_config(self):
config = super().get_config()
Expand Down

0 comments on commit 8cce221

Please sign in to comment.