File size: 406 Bytes
7057ade 1ffe419 7057ade |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
---
language: en
tags:
- google/mt5-small
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) |