未验证 提交 8cc4295e 编写于 作者: W Wei Shengyu 提交者: GitHub

Merge pull request #1907 from weisy11/cp-1906

Cherrypick #1906
...@@ -110,9 +110,8 @@ def init_model(config, net, optimizer=None): ...@@ -110,9 +110,8 @@ def init_model(config, net, optimizer=None):
load_distillation_model(net, pretrained_model) load_distillation_model(net, pretrained_model)
else: # common load else: # common load
load_dygraph_pretrain(net, path=pretrained_model) load_dygraph_pretrain(net, path=pretrained_model)
logger.info( logger.info("Finish load pretrained model from {}".format(
logger.coloring("Finish load pretrained model from {}".format( pretrained_model))
pretrained_model), "HEADER"))
def save_model(net, def save_model(net,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册