Update README.md
Browse files
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`)
|