readme: ---
  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)
  