mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Sneha DasandGitHub
19059da816
* feat(examples): align speech and audio examples with MNIST TrainJob flow This commit updates the speech-recognition and audio-classification notebooks to consistently use the Kubeflow Trainer SDK for TrainJob creation, status checking, and log streaming. It also adjusts the test workflows. Signed-off-by: Sneha Das <[email protected]> * fix: resolve torchcodec dependency and ddp timeout Signed-off-by: Sneha Das <[email protected]> * fix: align installation steps with mnist example Signed-off-by: Sneha Das <[email protected]> * fix: align installation step in speech recognition with mnist example Signed-off-by: Sneha Das <[email protected]> * fix: update SDK installation and paths in audio/speech notebooks Signed-off-by: Sneha Das <[email protected]> * chore: fix end of file newlines in notebooks Signed-off-by: Sneha Das <[email protected]> * fix: updated the installation path Signed-off-by: Sneha Das <[email protected]> --------- Signed-off-by: Sneha Das <[email protected]>
886 B
886 B
PyTorch Examples
This directory contains examples for training PyTorch models using the Kubeflow Trainer SDK.
Examples
| Task | Model | Dataset | Notebook |
|---|---|---|---|
| Image Classification | CNN | Fashion MNIST | mnist.ipynb |
| Question Answering | DistilBERT | SQuAD | fine-tune-distilbert.ipynb |
| Speech Recognition | Transformer | Speech Commands | speech-recognition.ipynb |
| Audio Classification | CNN (M5) | GTZAN | audio-classification.ipynb |
| Data Caching | [Model] | [Dataset] | data-cache-example.ipynb |
Prerequisites
To run these examples, install the Kubeflow SDK:
pip install -U kubeflow