diff --git a/paddlehub/finetune/task/basic_task.py b/paddlehub/finetune/task/basic_task.py index 758286a88b6a9a047c4b059267a7e304a47d15fd..c53504e874a3634105f8ecd9311a70347bc7e141 100644 --- a/paddlehub/finetune/task/basic_task.py +++ b/paddlehub/finetune/task/basic_task.py @@ -261,7 +261,7 @@ class BasicTask(object): var = self.env.main_program.global_block().vars[var_name] var.persistable = True - # to avoid to print logger two times in result of the logger usage of paddle-fluid + # to avoid to print logger two times in result of the logger usage of paddle-fluid 1.6 for handler in logging.root.handlers[:]: logging.root.removeHandler(handler) @@ -291,6 +291,10 @@ class BasicTask(object): self.exe.run(self.env.startup_program) + # to avoid to print logger two times in result of the logger usage of paddle-fluid 1.5 + for handler in logging.root.handlers[:]: + logging.root.removeHandler(handler) + self._build_env_end_event() @property @@ -588,6 +592,7 @@ class BasicTask(object): return self.finetune(do_eval=True) def finetune(self, do_eval=False): + # Start to finetune with self.phase_guard(phase="train"): self.init_if_necessary()