fix some doc errors
@@ -31,8 +31,6 @@ class ModelLoader(BaseLoader):
class ModelSaver(object):
"""Save a model
-
- :param str save_path: the path to the saving directory.
Example::
saver = ModelSaver("./save/model_ckpt_100.pkl")
@@ -41,6 +39,10 @@ class ModelSaver(object):
"""
def __init__(self, save_path):
+ """
+
+ :param save_path: the path to the saving directory.
+ """
self.save_path = save_path
def save_pytorch(self, model, param_only=True):
