paddlespeech/audiotools/ml/basemodel.py

paddlespeech/audiotools/ml/basemodel.py

@@ -110,7 +110,8 @@ class BaseModel(nn.Layer):
             state_dict = {"state_dict": self.state_dict(), "metadata": metadata}
             paddle.save(state_dict, str(path))
         else:
-            self._save_package(path, intern=intern, extern=extern, mock=mock)
+            raise NotImplementedError(
+                "Currently Paddle does not support packaging")
 
         return path
 
@@ -151,9 +152,6 @@ class BaseModel(nn.Layer):
         BaseModel
             A model that inherits from BaseModel.
         """
-        try:
-            model = cls._load_package(location, package_name=package_name)
-        except:
         model_dict = paddle.load(location)
         metadata = model_dict["metadata"]
         metadata["kwargs"].update(kwargs)
@@ -170,13 +168,6 @@ class BaseModel(nn.Layer):
 
         return model
 
-    def _save_package(self, path, intern=[], extern=[], mock=[], **kwargs):
-        raise NotImplementedError("Currently Paddle does not support packaging")
-
-    @classmethod
-    def _load_package(cls, path, package_name=None):
-        raise NotImplementedError("Currently Paddle does not support packaging")
-
     def save_to_folder(
             self,
             folder: typing.Union[str, Path],