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
381d88f6
Unverified
Commit
381d88f6
authored
Sep 12, 2018
by
Sami Kama
Committed by
GitHub
Sep 12, 2018
Browse files
Update logger.py
Spaces after commas
parent
eaaf891c
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
official/utils/logs/logger.py
official/utils/logs/logger.py
+2
-2
No files found.
official/utils/logs/logger.py
View file @
381d88f6
...
...
@@ -315,7 +315,7 @@ def _gather_run_info(model_name, dataset_name, run_params, test_id):
_collect_tensorflow_environment_variables
(
run_info
)
_collect_run_params
(
run_info
,
run_params
)
_collect_cpu_info
(
run_info
)
_collect_gpu_info
(
run_info
,
session_config
)
_collect_gpu_info
(
run_info
,
session_config
)
_collect_memory_info
(
run_info
)
_collect_test_environment
(
run_info
)
return
run_info
...
...
@@ -389,7 +389,7 @@ def _collect_cpu_info(run_info):
tf
.
logging
.
warn
(
"'cpuinfo' not imported. CPU info will not be logged."
)
def
_collect_gpu_info
(
run_info
,
session_config
=
None
):
def
_collect_gpu_info
(
run_info
,
session_config
=
None
):
"""Collect local GPU information by TF device library."""
gpu_info
=
{}
local_device_protos
=
device_lib
.
list_local_devices
(
session_config
)
...
...
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