revert comment

ce62ac15 · srihari-humbarwadi · add9b37e · ce62ac15
Unverified Commit ce62ac15 authored Oct 13, 2021 by srihari-humbarwadi
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/modeling/heads/segmentation_heads.py official/vision/beta/modeling/heads/segmentation_heads.py +1 -1

No files found.
--- a/official/vision/beta/modeling/heads/segmentation_heads.py
+++ b/official/vision/beta/modeling/heads/segmentation_heads.py
@@ -21,7 +21,7 @@ from official.vision.beta.modeling.layers import nn_layers
 from official.vision.beta.ops import spatial_transform_ops


-# @tf.keras.utils.register_keras_serializable(package='Vision')
+@tf.keras.utils.register_keras_serializable(package='Vision')
 class SegmentationHead(tf.keras.layers.Layer):
  """Creates a segmentation head."""