Commit
·
5a19c76
1
Parent(s):
c9f22c9
Upload README.md
Browse files
README.md
CHANGED
@@ -11,9 +11,9 @@ tags:
|
|
11 |
- t5
|
12 |
- mt5
|
13 |
datasets:
|
14 |
-
- wikipedia
|
15 |
metrics:
|
16 |
-
- accuracy
|
17 |
---
|
18 |
|
19 |
# mT5-base-nsp
|
@@ -72,6 +72,9 @@ encoded_dict = tokenizer.batch_encode_plus(batch_text_or_text_pairs=batch_texts,
|
|
72 |
print(torch.argmax(model(encoded_dict.input_ids, attention_mask=encoded_dict.attention_mask), dim=-1))
|
73 |
```
|
74 |
|
|
|
|
|
|
|
75 |
## BibTeX entry and citation info
|
76 |
|
77 |
```bibtex
|
@@ -84,4 +87,4 @@ print(torch.argmax(model(encoded_dict.input_ids, attention_mask=encoded_dict.att
|
|
84 |
}
|
85 |
```
|
86 |
|
87 |
-
The work is done with Ludwig-Maximilians-Universität Statistics group, don't forget to check out [their huggingface page](https://huggingface.co/misoda) for other interesting works!
|
|
|
11 |
- t5
|
12 |
- mt5
|
13 |
datasets:
|
14 |
+
- wikipedia
|
15 |
metrics:
|
16 |
+
- accuracy
|
17 |
---
|
18 |
|
19 |
# mT5-base-nsp
|
|
|
72 |
print(torch.argmax(model(encoded_dict.input_ids, attention_mask=encoded_dict.attention_mask), dim=-1))
|
73 |
```
|
74 |
|
75 |
+
### Training Metrics
|
76 |
+
<img src="https://huggingface.co/tolga-ozturk/mt5-base-nsp/resolve/main/metrics.png">
|
77 |
+
|
78 |
## BibTeX entry and citation info
|
79 |
|
80 |
```bibtex
|
|
|
87 |
}
|
88 |
```
|
89 |
|
90 |
+
The work is done with Ludwig-Maximilians-Universität Statistics group, don't forget to check out [their huggingface page](https://huggingface.co/misoda) for other interesting works!
|