Update README.md
Browse files
README.md
CHANGED
|
@@ -96,7 +96,7 @@ from uniem.finetuner import FineTuner
|
|
| 96 |
|
| 97 |
dataset = load_dataset('shibing624/nli_zh', 'STS-B')
|
| 98 |
# 指定训练的模型为 m3e-small
|
| 99 |
-
finetuner = FineTuner('moka-ai/m3e-small', dataset=dataset)
|
| 100 |
finetuner.run(epochs=1)
|
| 101 |
```
|
| 102 |
|
|
|
|
| 96 |
|
| 97 |
dataset = load_dataset('shibing624/nli_zh', 'STS-B')
|
| 98 |
# 指定训练的模型为 m3e-small
|
| 99 |
+
finetuner = FineTuner.from_pretrained('moka-ai/m3e-small', dataset=dataset)
|
| 100 |
finetuner.run(epochs=1)
|
| 101 |
```
|
| 102 |
|