sumandeng
/

my-m3e-base

sentence-transformers

Model card Files Files and versions Community

MokaHR commited on Jul 14, 2023

Commit

eec5494

•

1 Parent(s): 9714c5c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -96,7 +96,7 @@ from uniem.finetuner import FineTuner
 dataset = load_dataset('shibing624/nli_zh', 'STS-B')
 # 指定训练的模型为 m3e-small
-finetuner = FineTuner('moka-ai/m3e-small', dataset=dataset)
 finetuner.run(epochs=1)
 ```

 dataset = load_dataset('shibing624/nli_zh', 'STS-B')
 # 指定训练的模型为 m3e-small
+finetuner = FineTuner.from_pretrained('moka-ai/m3e-small', dataset=dataset)
 finetuner.run(epochs=1)
 ```