mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Andrey VelichkevichandGitHub
fbacd9f577
* fix(examples): Verify TrainJob Completion Signed-off-by: Andrey Velichkevich <[email protected]> * Rename client to TrainerClient() Signed-off-by: Andrey Velichkevich <[email protected]> --------- Signed-off-by: Andrey Velichkevich <[email protected]>
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