Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,10 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
|
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("flax-community/roberta-swahili")
|
| 19 |
|
| 20 |
model = AutoModelForMaskedLM.from_pretrained("flax-community/roberta-swahili")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
```
|
| 22 |
|
| 23 |
#### **Training Data**:
|
|
|
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained("flax-community/roberta-swahili")
|
| 19 |
|
| 20 |
model = AutoModelForMaskedLM.from_pretrained("flax-community/roberta-swahili")
|
| 21 |
+
|
| 22 |
+
print(round((model.num_parameters())/(1000*1000)),"Million Parameters")
|
| 23 |
+
|
| 24 |
+
105 Million Parameters
|
| 25 |
```
|
| 26 |
|
| 27 |
#### **Training Data**:
|