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)