snehalyelmati
commited on
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)
|