mirror of
https://github.com/hanzo-ml/trainer.git
synced 2026-09-24 00:31:34 +00:00
Yash AgarwalandGitHub
0f4c4be96d
* fix(examples): use namespaced SQuAD dataset Signed-off-by: XploY04 <[email protected]> * fix(runtimes): cap kagglehub below 1.0 Signed-off-by: XploY04 <[email protected]> --------- Signed-off-by: XploY04 <[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