Przeglądaj źródła

refine info message

will-jl944 4 lat temu
rodzic
commit
ea8b3d78fd
1 zmienionych plików z 1 dodań i 1 usunięć
  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)