Skip to content

Commit e52126f

Browse files
Steffy-zxfnepeplwu
authored andcommitted
fix logger when using paddle 1.5/1.6 (#252)
1 parent 1152db5 commit e52126f

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

paddlehub/finetune/task/basic_task.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def _build_env(self):
261261
var = self.env.main_program.global_block().vars[var_name]
262262
var.persistable = True
263263

264-
# to avoid to print logger two times in result of the logger usage of paddle-fluid
264+
# to avoid to print logger two times in result of the logger usage of paddle-fluid 1.6
265265
for handler in logging.root.handlers[:]:
266266
logging.root.removeHandler(handler)
267267

@@ -291,6 +291,10 @@ def _build_env(self):
291291

292292
self.exe.run(self.env.startup_program)
293293

294+
# to avoid to print logger two times in result of the logger usage of paddle-fluid 1.5
295+
for handler in logging.root.handlers[:]:
296+
logging.root.removeHandler(handler)
297+
294298
self._build_env_end_event()
295299

296300
@property
@@ -588,6 +592,7 @@ def finetune_and_eval(self):
588592
return self.finetune(do_eval=True)
589593

590594
def finetune(self, do_eval=False):
595+
591596
# Start to finetune
592597
with self.phase_guard(phase="train"):
593598
self.init_if_necessary()

0 commit comments

Comments
 (0)