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

add default args

Browse Source
This commit is contained in:
雨泓
2022-06-22 21:54:41 +08:00
parent 3987a3bf7d
commit 96e25be7d2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/models/nlp/sbert_for_sentence_similarity.py
View File

@@ -18,6 +18,6 @@ class SbertForSentenceSimilarity(SbertForSequenceClassificationBase):
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, **kwargs)
super().__init__(model_dir, *args, model_args={"num_labels": 2}, **kwargs)
self.model_dir = model_dir
assert self.model.config.num_labels == 2

2
modelscope/models/nlp/sbert_for_sentiment_classification.py
View File

@@ -19,5 +19,5 @@ class SbertForSentimentClassification(SbertForSequenceClassificationBase):
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, **kwargs)
super().__init__(model_dir, *args, model_args={"num_labels": 2}, **kwargs)
assert self.model.config.num_labels == 2