Skip to content

Commit a26d899

Browse files
authored
hf dl example (#260)
1 parent 789e322 commit a26d899

2 files changed

Lines changed: 640 additions & 0 deletions

File tree

notebooks/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,4 @@ a few minutes.
77
| Notebook | What it shows | Open |
88
|---|---|---|
99
| `data_loading_bottleneck.ipynb` | Diagnose and fix a data-loading (input-bound) bottleneck on a real ResNet-18 + Imagenette run: train twice, change only the DataLoader, and read the before/after wall-clock speedup and GPU utilization from TraceML | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/traceopt-ai/traceml/blob/main/notebooks/data_loading_bottleneck.ipynb) |
10+
| `huggingface_dataloading_bottleneck.ipynb` | The Hugging Face Trainer path: diagnose and fix the same real-image input bottleneck with `TraceMLTrainer`, ResNet-50, and Imagenette by changing only `TrainingArguments` data-loader settings | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/traceopt-ai/traceml/blob/main/notebooks/huggingface_dataloading_bottleneck.ipynb) |

0 commit comments

Comments
 (0)