Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 03:59:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
ly119399
2022-06-30 15:26:19 +08:00
parent 94522aa721 f84fe9bf24
commit 99c013acc3
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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)