File size: 603 Bytes
7057ade
 
 
1ffe419
44f8d45
7057ade
 
 
 
 
 
 
 
 
 
 
 
d5a2c3c
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
---
language: en
tags:
- google/mt5-small
- machine_translation
license: apache-2.0
---

# Hindi-English Translation Model

Based on the "google/mt5-small" pre-trained model. Fine-tuned it on Hindi to English dataset. 

### Parameters
- number of epochs = 8
- batch size = 16
- learning rate = 5e-4
- number of batches = int(np.ceil(len(dataset) / batch size))
- n_warmup_steps = int(number of epochs * number of batches * 0.01)

### Training Loss

![loss.png](./h2e_loss.png)

### Examples

![example_1.png](./h2e_example1.png)
![example_2.png](./h2e_example2.png)
![example_3.png](./h2e_example3.png)