diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index 625a2848339e47e688a263631cfe4a5b9e6c691b..c0125ec49f18dc9886fe2df9852d481d8463705a 100644 --- a/ppcls/utils/save_load.py +++ b/ppcls/utils/save_load.py @@ -110,9 +110,8 @@ def init_model(config, net, optimizer=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,