Skip to content

GitLab

Commit 2a595d51 authored by Yeqing Li's avatar Yeqing Li Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 410934847
parent eea839c4
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
official/vision/detection/modeling/architecture/nn_blocks.py
View file @ 2a595d51
...@@ -23,7 +23,6 @@ import tensorflow as tf ...@@ -23,7 +23,6 @@ import tensorflow as tf
from official.modeling import tf_utils from official.modeling import tf_utils
@tf.keras.utils.register_keras_serializable(package='Vision')
class ResidualBlock(tf.keras.layers.Layer): class ResidualBlock(tf.keras.layers.Layer):
"""A residual block.""" """A residual block."""
...@@ -163,7 +162,6 @@ class ResidualBlock(tf.keras.layers.Layer): ...@@ -163,7 +162,6 @@ class ResidualBlock(tf.keras.layers.Layer):
return self._activation_fn(x + shortcut) return self._activation_fn(x + shortcut)
@tf.keras.utils.register_keras_serializable(package='Vision')
class BottleneckBlock(tf.keras.layers.Layer): class BottleneckBlock(tf.keras.layers.Layer):
"""A standard bottleneck block.""" """A standard bottleneck block."""
......
official/vision/detection/modeling/architecture/spinenet.py
View file @ 2a595d51
...@@ -113,7 +113,6 @@ def build_block_specs(block_specs=None): ...@@ -113,7 +113,6 @@ def build_block_specs(block_specs=None):
return [BlockSpec(*b) for b in block_specs] return [BlockSpec(*b) for b in block_specs]
@tf.keras.utils.register_keras_serializable(package='Vision')
class SpineNet(tf.keras.Model): class SpineNet(tf.keras.Model):
"""Class to build SpineNet models.""" """Class to build SpineNet models."""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment