Browse Source

Merge branch 'feat/nlp' into nlp/space/dst

master
ly119399 3 years ago
parent
commit
99c013acc3
1 changed files with 0 additions and 2 deletions
  1. +0
    -2
      modelscope/models/nlp/sbert_for_sentiment_classification.py

+ 0
- 2
modelscope/models/nlp/sbert_for_sentiment_classification.py View File

@@ -16,8 +16,6 @@ class SbertForSentimentClassification(SbertForSequenceClassificationBase):

Args:
model_dir (str): the model path.
model_cls (Optional[Any], optional): model loader, if None, use the
default loader to load model weights, by default None.
"""
super().__init__(
model_dir, *args, model_args={'num_labels': 2}, **kwargs)


Loading…
Cancel
Save