Commit
•
7057ade
1
Parent(s):
80f4c5c
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- exbert
|
| 5 |
+
license: apache-2.0
|
| 6 |
+
---
|
| 7 |
+
|
| 8 |
+
# Hindi-English Translation Model
|
| 9 |
+
|
| 10 |
+
Based on the "google/mt5-small" pre-trained model. Fine-tuned it on Hindi to English dataset.
|
| 11 |
+
|
| 12 |
+
### Parameters
|
| 13 |
+
- number of epochs = 8
|
| 14 |
+
- batch size = 16
|
| 15 |
+
- learning rate = 5e-4
|
| 16 |
+
- number of batches = int(np.ceil(len(dataset) / batch size))
|
| 17 |
+
- n_warmup_steps = int(number of epochs * number of batches * 0.01)
|