diff --git a/README.md b/README.md index d37c853..830d27a 100644 --- a/README.md +++ b/README.md @@ -331,6 +331,7 @@ and then return text embedding in ndarray. ​ The text embedding extracted by model. +
***save_model(format='pytorch', path='default')*** @@ -342,10 +343,11 @@ Save model to local with specified format. ​ The format of saved model, defaults to 'pytorch'. -***format***: *path* +***path***: *str* ​ The path where model is saved to. By default, it will save model to the operator directory. +
***supported_model_names(format=None)***