浏览代码

refine info message

will-jl944 4 年之前
父节点
当前提交
ea8b3d78fd
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dygraph/paddlex/cv/models/base.py

+ 1 - 1
dygraph/paddlex/cv/models/base.py

@@ -84,7 +84,7 @@ class BaseModel:
                 load_pretrain_weights(
                     self.net.backbone,
                     pretrain_weights,
-                    model_name=self.model_name)
+                    model_name='backbone of ' + self.model_name)
             else:
                 load_pretrain_weights(
                     self.net, pretrain_weights, model_name=self.model_name)