Beta
Explore
Marketplace
Neural Labs
Chat
Wallet
Docs
transformer-en-hi – AI Model by m25csa001 | AlphaNeural AI
You can deploy this model and start earning money today!
m25csa001
/
transformer-en-hi
like
0
translation
english-to-hindi
transformer
pytorch
hi
us
Views
No views yet
Model card
Files and Versions
Community
API
Deploy
English to Hindi Transformer
Assignment 4 — Optimizing Transformer Translation with Ray Tune & Optuna
Model Details
Architecture: Custom Transformer (from scratch)
d_model: 512 | num_heads: 8 | d_ff: 2048 | num_layers: 4
Best lr: 2.037e-4 | dropout: 0.102 | batch_size: 64
Tuned with: Ray Tune + OptunaSearch + ASHAScheduler
Results
Metric
Baseline (100 epochs)
Tuned (40 epochs)
BLEU Score
52.47
73.70
Final Loss
0.0974
0.0805
Training Time
~52 min
4.4 min
GitHub Repo
https://github.com/yadityax/MLOps-Aditya_Arun_Kumar_Yadav-M25CSA001/tree/Assignment-4
