DeepakKumarMSL commited on
Commit
5ce3128
·
verified ·
1 Parent(s): c7f2b64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -5,6 +5,7 @@ This repository contains a **quantized English-to-Hindi translation model** fine
5
  ## 🔧 Model Details
6
 
7
  - **Base model**: [`Helsinki-NLP/opus-mt-en-hi`](https://huggingface.co/Helsinki-NLP/opus-mt-en-hi)
 
8
  - **Training platform**: Kaggle (CUDA GPU)
9
  - **Fine-tuned**: On English-Hindi pairs from the Hugging Face dataset
10
  - **Quantization**: PyTorch Dynamic Quantization (`torch.quantization.quantize_dynamic`)
 
5
  ## 🔧 Model Details
6
 
7
  - **Base model**: [`Helsinki-NLP/opus-mt-en-hi`](https://huggingface.co/Helsinki-NLP/opus-mt-en-hi)
8
+ - **Dataset**: Aarif1430/english-to-hindi
9
  - **Training platform**: Kaggle (CUDA GPU)
10
  - **Fine-tuned**: On English-Hindi pairs from the Hugging Face dataset
11
  - **Quantization**: PyTorch Dynamic Quantization (`torch.quantization.quantize_dynamic`)