Internal change

PiperOrigin-RevId: 410934847

official/vision/detection/modeling/architecture/nn_blocks.py

@@ -23,7 +23,6 @@ import tensorflow as tf
 from official.modeling import tf_utils
 
 
-@tf.keras.utils.register_keras_serializable(package='Vision')
 class ResidualBlock(tf.keras.layers.Layer):
   """A residual block."""
 
@@ -163,7 +162,6 @@ class ResidualBlock(tf.keras.layers.Layer):
     return self._activation_fn(x + shortcut)
 
 
-@tf.keras.utils.register_keras_serializable(package='Vision')
 class BottleneckBlock(tf.keras.layers.Layer):
   """A standard bottleneck block."""
 

official/vision/detection/modeling/architecture/spinenet.py

@@ -113,7 +113,6 @@ def build_block_specs(block_specs=None):
   return [BlockSpec(*b) for b in block_specs]
 
 
-@tf.keras.utils.register_keras_serializable(package='Vision')
 class SpineNet(tf.keras.Model):
   """Class to build SpineNet models."""