Skip to content

GitLab

Commit 36ab0686 authored by Yuqi Li's avatar Yuqi Li Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 398417598
parent e53d3255
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/beta/modeling/heads/segmentation_heads.py
View file @ 36ab0686
......@@ -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(
......
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