Skip to content

GitLab

Unverified Commit 381d88f6 authored by Sami Kama's avatar Sami Kama Committed by GitHub
Browse files

Update logger.py 

Spaces after commas
parent eaaf891c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/utils/logs/logger.py
View file @ 381d88f6
...@@ -315,7 +315,7 @@ def _gather_run_info(model_name, dataset_name, run_params, test_id): ...@@ -315,7 +315,7 @@ def _gather_run_info(model_name, dataset_name, run_params, test_id):
_collect_tensorflow_environment_variables(run_info) _collect_tensorflow_environment_variables(run_info)
_collect_run_params(run_info, run_params) _collect_run_params(run_info, run_params)
_collect_cpu_info(run_info) _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_memory_info(run_info)
_collect_test_environment(run_info) _collect_test_environment(run_info)
return run_info return run_info
...@@ -389,7 +389,7 @@ def _collect_cpu_info(run_info): ...@@ -389,7 +389,7 @@ def _collect_cpu_info(run_info):
tf.logging.warn("'cpuinfo' not imported. CPU info will not be logged.") 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.""" """Collect local GPU information by TF device library."""
gpu_info = {} gpu_info = {}
local_device_protos = device_lib.list_local_devices(session_config) local_device_protos = device_lib.list_local_devices(session_config)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment