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

Update logger.py

Spaces after commas
parent eaaf891c
......@@ -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)
......
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