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
2a26552c
Commit
2a26552c
authored
Dec 13, 2021
by
Liangzhe Yuan
Committed by
A. Unique TensorFlower
Dec 13, 2021
Browse files
#movinet set NormActivation.activation = None by default and gauge its use case.
PiperOrigin-RevId: 416126695
parent
fceb3ca9
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
3 deletions
+7
-3
official/vision/beta/projects/movinet/configs/movinet.py
official/vision/beta/projects/movinet/configs/movinet.py
+1
-1
official/vision/beta/projects/movinet/modeling/movinet.py
official/vision/beta/projects/movinet/modeling/movinet.py
+6
-2
No files found.
official/vision/beta/projects/movinet/configs/movinet.py
View file @
2a26552c
...
...
@@ -126,7 +126,7 @@ class MovinetModel(video_classification.VideoClassificationModel):
model_type
:
str
=
'movinet'
backbone
:
Backbone3D
=
Backbone3D
()
norm_activation
:
common
.
NormActivation
=
common
.
NormActivation
(
activation
=
'swish'
,
activation
=
None
,
# legacy flag, not used.
norm_momentum
=
0.99
,
norm_epsilon
=
1e-3
,
use_sync_bn
=
True
)
...
...
official/vision/beta/projects/movinet/modeling/movinet.py
View file @
2a26552c
...
...
@@ -709,8 +709,12 @@ def build_movinet(
"""Builds MoViNet backbone from a config."""
backbone_type
=
backbone_config
.
type
backbone_cfg
=
backbone_config
.
get
()
assert
backbone_type
==
'movinet'
,
(
'Inconsistent backbone type '
f
'
{
backbone_type
}
'
)
if
backbone_type
!=
'movinet'
:
raise
ValueError
(
f
'Inconsistent backbone type
{
backbone_type
}
'
)
if
norm_activation_config
.
activation
is
not
None
:
raise
ValueError
(
'norm_activation is not used in MoViNets, but specified: %s'
%
norm_activation_config
.
activation
)
return
Movinet
(
model_id
=
backbone_cfg
.
model_id
,
...
...
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