Internal change

PiperOrigin-RevId: 398417598

Internal change
PiperOrigin-RevId: 398417598
36ab0686 · Yuqi Li · A. Unique TensorFlower · e53d3255 · 36ab0686
Commit 36ab0686 authored Sep 22, 2021 by Yuqi Li Committed by A. Unique TensorFlower Sep 22, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/vision/beta/modeling/heads/segmentation_heads.py official/vision/beta/modeling/heads/segmentation_heads.py +2 -0

No files found.
--- a/official/vision/beta/modeling/heads/segmentation_heads.py
+++ b/official/vision/beta/modeling/heads/segmentation_heads.py
@@ -155,6 +155,8 @@ class SegmentationHead(tf.keras.layers.Layer):
                depthwise_initializer=random_initializer,
                depthwise_regularizer=self._config_dict['kernel_regularizer'],
                depth_multiplier=1))
+        norm_name = 'segmentation_head_depthwise_norm_{}'.format(i)
+        self._norms.append(bn_op(name=norm_name, **bn_kwargs))
      conv_name = 'segmentation_head_conv_{}'.format(i)
      self._convs.append(
          conv_op(