From f6923adff82b47bb8352473e42a362050950b220 Mon Sep 17 00:00:00 2001
From: weishengyu <weisy11@163.com>
Date: Tue, 10 May 2022 14:29:20 +0800
Subject: [PATCH] remove logger.colring

---
 ppcls/utils/save_load.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py
index 09325537..4e27f12c 100644
--- a/ppcls/utils/save_load.py
+++ b/ppcls/utils/save_load.py
@@ -116,9 +116,8 @@ def init_model(config, net, optimizer=None, loss: paddle.nn.Layer=None):
             load_distillation_model(net, pretrained_model)
         else:  # common load
             load_dygraph_pretrain(net, path=pretrained_model)
-            logger.info(
-                logger.coloring("Finish load pretrained model from {}".format(
-                    pretrained_model), "HEADER"))
+            logger.info("Finish load pretrained model from {}".format(
+                    pretrained_model))
 
 
 def save_model(net,
-- 
GitLab