Skip to content

Commit 2de1029

Browse files
xiegegegeccmeteorljh
authored andcommitted
fix logger problem (#21342) (#21635)
* fix logger problem test=develop * refine logger test=develop
1 parent f7c629d commit 2de1029

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

python/paddle/fluid/trainer_factory.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@
1818
import logging
1919
import numpy as np
2020

21-
logging.basicConfig()
21+
FORMAT = '%(asctime)s-%(levelname)s: %(message)s'
22+
logging.basicConfig(level=logging.INFO, format=FORMAT)
23+
local_logger = logging.getLogger(__name__)
2224

2325
from .trainer_desc import MultiTrainer, DistMultiTrainer, PipelineTrainer
2426
from .device_worker import Hogwild, DownpourSGD, Section
@@ -109,7 +111,8 @@ def handler_launch_func(self, scope, handler):
109111
if isinstance(fetch_instance.var_dict[key], Variable):
110112
var_name_to_key[fetch_instance.var_dict[key].name] = key
111113
else:
112-
logging.warning("the value of {} is not a Variable".format(key))
114+
local_logger.warning("the value of {} is not a Variable".format(
115+
key))
113116
var_name_to_key["None.var"] = key
114117
elapsed_secs = 0
115118
while True:
@@ -127,8 +130,8 @@ def handler_launch_func(self, scope, handler):
127130
var = scope.find_var(key)
128131
fetch_dict[key] = var
129132
if var == None:
130-
logging.warning("{} value currently not available".
131-
format(var_name_to_key[key]))
133+
local_logger.warning("{} value currently not available".
134+
format(var_name_to_key[key]))
132135
res_dict = {}
133136
for key in fetch_dict:
134137
user_name = var_name_to_key[key]

0 commit comments

Comments
 (0)