提交 9985d123 编写于 作者: S Scott Zhu 提交者: A. Unique TensorFlower

Prepare for upcoming keras initializer change.

PiperOrigin-RevId: 451510454
上级 2397c70d
...@@ -257,12 +257,12 @@ class DeepMaskHead(tf.keras.layers.Layer): ...@@ -257,12 +257,12 @@ class DeepMaskHead(tf.keras.layers.Layer):
variant = self._config_dict['convnet_variant'] variant = self._config_dict['convnet_variant']
if variant == 'default': if variant == 'default':
conv_op, conv_kwargs = self._get_conv_op_and_kwargs()
bn_op, bn_kwargs = self._get_bn_op_and_kwargs() bn_op, bn_kwargs = self._get_bn_op_and_kwargs()
self._convs = [] self._convs = []
self._conv_norms = [] self._conv_norms = []
for i in range(self._config_dict['num_convs']): for i in range(self._config_dict['num_convs']):
conv_name = 'mask-conv_{}'.format(i) conv_name = 'mask-conv_{}'.format(i)
conv_op, conv_kwargs = self._get_conv_op_and_kwargs()
self._convs.append(conv_op(name=conv_name, **conv_kwargs)) self._convs.append(conv_op(name=conv_name, **conv_kwargs))
bn_name = 'mask-conv-bn_{}'.format(i) bn_name = 'mask-conv-bn_{}'.format(i)
self._conv_norms.append(bn_op(name=bn_name, **bn_kwargs)) self._conv_norms.append(bn_op(name=bn_name, **bn_kwargs))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册