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
6b2bc083
Commit
6b2bc083
authored
Aug 19, 2020
by
A. Unique TensorFlower
Browse files
Log encoder scaffold arguments.
PiperOrigin-RevId: 327518640
parent
0fddab49
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
0 deletions
+3
-0
official/nlp/modeling/networks/encoder_scaffold.py
official/nlp/modeling/networks/encoder_scaffold.py
+3
-0
No files found.
official/nlp/modeling/networks/encoder_scaffold.py
View file @
6b2bc083
...
@@ -22,6 +22,7 @@ from __future__ import print_function
...
@@ -22,6 +22,7 @@ from __future__ import print_function
import
inspect
import
inspect
from
absl
import
logging
import
gin
import
gin
import
tensorflow
as
tf
import
tensorflow
as
tf
...
@@ -205,6 +206,8 @@ class EncoderScaffold(tf.keras.Model):
...
@@ -205,6 +206,8 @@ class EncoderScaffold(tf.keras.Model):
super
(
EncoderScaffold
,
self
).
__init__
(
super
(
EncoderScaffold
,
self
).
__init__
(
inputs
=
inputs
,
outputs
=
outputs
,
**
kwargs
)
inputs
=
inputs
,
outputs
=
outputs
,
**
kwargs
)
logging
.
info
(
'EncoderScaffold configs: %s'
,
self
.
get_config
())
def
get_config
(
self
):
def
get_config
(
self
):
config_dict
=
{
config_dict
=
{
'num_hidden_instances'
:
self
.
_num_hidden_instances
,
'num_hidden_instances'
:
self
.
_num_hidden_instances
,
...
...
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