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
d5722dcd
Commit
d5722dcd
authored
Aug 21, 2019
by
Vinh Nguyen
Browse files
minor fix to comments
parent
c9c05e9b
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
official/transformer/v2/transformer_main.py
official/transformer/v2/transformer_main.py
+1
-1
official/vision/image_classification/resnet_imagenet_main.py
official/vision/image_classification/resnet_imagenet_main.py
+1
-1
No files found.
official/transformer/v2/transformer_main.py
View file @
d5722dcd
...
@@ -380,7 +380,7 @@ class TransformerTask(object):
...
@@ -380,7 +380,7 @@ class TransformerTask(object):
# Note: when flags_obj.fp16_implementation == "graph_rewrite",
# Note: when flags_obj.fp16_implementation == "graph_rewrite",
# dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32'
# dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32'
# which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite
# which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite
# do
es
not double up.
# do not double up.
opt
=
tf
.
train
.
experimental
.
enable_mixed_precision_graph_rewrite
(
opt
)
opt
=
tf
.
train
.
experimental
.
enable_mixed_precision_graph_rewrite
(
opt
)
return
opt
return
opt
...
...
official/vision/image_classification/resnet_imagenet_main.py
View file @
d5722dcd
...
@@ -188,7 +188,7 @@ def run(flags_obj):
...
@@ -188,7 +188,7 @@ def run(flags_obj):
# Note: when flags_obj.fp16_implementation == "graph_rewrite",
# Note: when flags_obj.fp16_implementation == "graph_rewrite",
# dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32'
# dtype as determined by flags_core.get_tf_dtype(flags_obj) would be 'float32'
# which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite
# which will ensure tf.keras.mixed_precision and tf.train.experimental.enable_mixed_precision_graph_rewrite
# do
es
not double up.
# do not double up.
optimizer
=
tf
.
train
.
experimental
.
enable_mixed_precision_graph_rewrite
(
optimizer
)
optimizer
=
tf
.
train
.
experimental
.
enable_mixed_precision_graph_rewrite
(
optimizer
)
if
flags_obj
.
use_trivial_model
:
if
flags_obj
.
use_trivial_model
:
...
...
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