Update README.md
Browse files
README.md
CHANGED
|
@@ -57,6 +57,9 @@ print(f"Distance: {cos_sim[0,1].detach().item()}")
|
|
| 57 |
|
| 58 |
```
|
| 59 |
|
|
|
|
|
|
|
|
|
|
| 60 |
|
| 61 |
# Benchmark
|
| 62 |
|
|
|
|
| 57 |
|
| 58 |
```
|
| 59 |
|
| 60 |
+
# Training data
|
| 61 |
+
|
| 62 |
+
The model was trained on a random collection of sentences from Wikipedia: [Training data file](https://huggingface.co/datasets/princeton-nlp/datasets-for-simcse/resolve/main/wiki1m_for_simcse.txt)
|
| 63 |
|
| 64 |
# Benchmark
|
| 65 |
|