Commit
•
d5a2c3c
1
Parent(s):
a9d85f6
Update README.md
Browse files
README.md
CHANGED
@@ -14,4 +14,14 @@ Based on the "google/mt5-small" pre-trained model. Fine-tuned it on Hindi to Eng
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|
18 |
+
|
19 |
+
### Training Loss
|
20 |
+
|
21 |
+
![loss.png](./h2e_loss.png)
|
22 |
+
|
23 |
+
### Examples
|
24 |
+
|
25 |
+
![example_1.png](./h2e_example1.png)
|
26 |
+
![example_2.png](./h2e_example2.png)
|
27 |
+
![example_3.png](./h2e_example3.png)
|