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

fix bug of modelscope.trainers.nlp.sentence_embedding_trainer.get_data_collator

Browse Source 
修正sentence_embedding_trainer 初始化bug

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11725227
This commit is contained in:
zhangyanzhao.zyz
2023-02-21 22:42:13 +08:00
committed by wenmeng.zwm
parent fb9c41092a
commit 2ebb6db260
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modelscope/trainers/nlp/sentence_embedding_trainer.py
View File

@@ -86,7 +86,7 @@ class SentenceEmbeddingTrainer(NlpEpochBasedTrainer):
model_revision=model_revision,
**kwargs)
def get_data_collator(self, data_collator):
def get_data_collator(self, data_collator, **kwargs):
"""Get the data collator for both training and evaluating.
Args:
@@ -99,7 +99,7 @@ class SentenceEmbeddingTrainer(NlpEpochBasedTrainer):
data_collator = SentenceEmbeddingCollator(
tokenizer=self.train_preprocessor.nlp_tokenizer,
max_length=self.train_preprocessor.max_length)
return super().get_data_collator(data_collator)
return super().get_data_collator(data_collator, **kwargs)
def evauate(self):
return {}