fix logger problem (#21342)

* fix logger problem
test=develop

* refine logger
test=develop
revert-21172-masked_select_api
Dong Daxiang 5 years ago committed by GitHub
parent 345b67b5e2
commit 0a93635b5f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -18,7 +18,9 @@ import time
import logging
import numpy as np
logging.basicConfig()
FORMAT = '%(asctime)s-%(levelname)s: %(message)s'
logging.basicConfig(level=logging.INFO, format=FORMAT)
local_logger = logging.getLogger(__name__)
from .trainer_desc import MultiTrainer, DistMultiTrainer, PipelineTrainer
from .device_worker import Hogwild, DownpourSGD, Section
@ -109,7 +111,8 @@ class FetchHandlerMonitor(object):
if isinstance(fetch_instance.var_dict[key], Variable):
var_name_to_key[fetch_instance.var_dict[key].name] = key
else:
logging.warning("the value of {} is not a Variable".format(key))
local_logger.warning("the value of {} is not a Variable".format(
key))
var_name_to_key["None.var"] = key
elapsed_secs = 0
while True:
@ -127,7 +130,7 @@ class FetchHandlerMonitor(object):
var = scope.find_var(key)
fetch_dict[key] = var
if var == None:
logging.warning("{} value currently not available".
local_logger.warning("{} value currently not available".
format(var_name_to_key[key]))
res_dict = {}
for key in fetch_dict:

Loading…
Cancel
Save