ML Training Pipeline
End-to-end machine learning pipeline for image classification using PyTorch and MLflow.
Designed and implemented a production-ready ML training pipeline with experiment tracking, model versioning, and automated retraining.
Stack
- PyTorch for model training
- MLflow for experiment tracking
- Docker for reproducible environments
- GitHub Actions for CI/CD