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
08af7775
Unverified
Commit
08af7775
authored
Mar 23, 2018
by
Qianli Scott Zhu
Committed by
GitHub
Mar 23, 2018
Browse files
Fix random order problem in benchmark logging. (#3725)
parent
b9b44f7b
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
7 additions
and
7 deletions
+7
-7
official/utils/logging/logger.py
official/utils/logging/logger.py
+1
-1
official/utils/logging/logger_test.py
official/utils/logging/logger_test.py
+6
-6
No files found.
official/utils/logging/logger.py
View file @
08af7775
...
...
@@ -52,7 +52,7 @@ class BenchmarkLogger(object):
type
(
eval_results
))
return
global_step
=
eval_results
[
tf
.
GraphKeys
.
GLOBAL_STEP
]
for
key
in
eval_results
:
for
key
in
sorted
(
eval_results
)
:
if
key
!=
tf
.
GraphKeys
.
GLOBAL_STEP
:
self
.
log_metric
(
key
,
eval_results
[
key
],
global_step
=
global_step
)
...
...
official/utils/logging/logger_test.py
View file @
08af7775
...
...
@@ -98,18 +98,18 @@ class BenchmarkLoggerTest(tf.test.TestCase):
metric_log
=
os
.
path
.
join
(
log_dir
,
"metric.log"
)
self
.
assertTrue
(
tf
.
gfile
.
Exists
(
metric_log
))
with
tf
.
gfile
.
GFile
(
metric_log
)
as
f
:
loss
=
json
.
loads
(
f
.
readline
())
self
.
assertEqual
(
loss
[
"name"
],
"loss"
)
self
.
assertEqual
(
loss
[
"value"
],
0.46237424
)
self
.
assertEqual
(
loss
[
"unit"
],
None
)
self
.
assertEqual
(
loss
[
"global_step"
],
207082
)
accuracy
=
json
.
loads
(
f
.
readline
())
self
.
assertEqual
(
accuracy
[
"name"
],
"accuracy"
)
self
.
assertEqual
(
accuracy
[
"value"
],
0.9285
)
self
.
assertEqual
(
accuracy
[
"unit"
],
None
)
self
.
assertEqual
(
accuracy
[
"global_step"
],
207082
)
loss
=
json
.
loads
(
f
.
readline
())
self
.
assertEqual
(
loss
[
"name"
],
"loss"
)
self
.
assertEqual
(
loss
[
"value"
],
0.46237424
)
self
.
assertEqual
(
loss
[
"unit"
],
None
)
self
.
assertEqual
(
loss
[
"global_step"
],
207082
)
def
test_log_evaluation_result_with_invalid_type
(
self
):
eval_result
=
"{'loss': 0.46237424, 'global_step': 207082}"
log_dir
=
tempfile
.
mkdtemp
(
dir
=
self
.
get_temp_dir
())
...
...
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