diff --git a/ppcls/utils/save_load.py b/ppcls/utils/save_load.py index 093255379cd35875fbaf06282e391017bf7f14a3..4e27f12c1d4830f2f16580bfa976cf3ace78d934 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,