Merge pull request #1906 from weisy11/fix-logger

[bug fix]remove logger.colring
2022-05-10 15:34:05 +08:00 · 2022-05-10 15:34:05 +08:00 · 4411268755
parent 283ae9b327 f6923adff8
commit 4411268755
1 changed files with 2 additions and 3 deletions
--- 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,