Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
9985d123
Commit
9985d123
authored
May 27, 2022
by
Scott Zhu
Committed by
A. Unique TensorFlower
May 27, 2022
Browse files
Prepare for upcoming keras initializer change.
PiperOrigin-RevId: 451510454
parent
2397c70d
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
official/projects/deepmac_maskrcnn/modeling/heads/instance_heads.py
...rojects/deepmac_maskrcnn/modeling/heads/instance_heads.py
+1
-1
No files found.
official/projects/deepmac_maskrcnn/modeling/heads/instance_heads.py
View file @
9985d123
...
...
@@ -257,12 +257,12 @@ class DeepMaskHead(tf.keras.layers.Layer):
variant
=
self
.
_config_dict
[
'convnet_variant'
]
if
variant
==
'default'
:
conv_op
,
conv_kwargs
=
self
.
_get_conv_op_and_kwargs
()
bn_op
,
bn_kwargs
=
self
.
_get_bn_op_and_kwargs
()
self
.
_convs
=
[]
self
.
_conv_norms
=
[]
for
i
in
range
(
self
.
_config_dict
[
'num_convs'
]):
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
))
bn_name
=
'mask-conv-bn_{}'
.
format
(
i
)
self
.
_conv_norms
.
append
(
bn_op
(
name
=
bn_name
,
**
bn_kwargs
))
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment