How do I fix memory leaks when fine-tuning open source models on AWS SageMaker because our training job keeps crashing
Download run
Fan-out diagram
The searches the AI actually ran for this prompt.
SEED QUERY
How do I fix memory leaks when fine-tuning open source models on AWS SageMaker because our training job keeps crashing
3 real searches · 1 real-search pass
AWS SageMaker training jobs troubleshooting out of memory GPU memory PyTorch official documentationGoogle ↗
OBSERVED
PyTorch CUDA memory management memory_snapshot empty_cache official docsGoogle ↗
OBSERVED
Hugging Face Transformers performance gradient checkpointing evaluation memory official docsGoogle ↗
OBSERVED
Badge and edge weight = share of passes that produced the query. Hover a query for its intent and routing format.
Query classes
Google's semantic query class for every sub-query: what kind of answer each search is after, simulated vs real.
Classifying 3 queries…
Searches and sources
ChatGPT3 real searches · gpt-6
AWS SageMaker training jobs troubleshooting out of memory GPU memory PyTorch official documentation Google ↗
PyTorch CUDA memory management memory_snapshot empty_cache official docs Google ↗
Hugging Face Transformers performance gradient checkpointing evaluation memory official docs Google ↗
- Troubleshoot errors when you run SageMaker AI training jobs · repost.aws
- Model training · docs.aws.amazon.com
- CUDA semantics · docs.pytorch.org
- Understanding CUDA Memory Usage · docs.pytorch.org
- PyTorch CUDA Memory Visualizer · pytorch.org
Queries
3 queries · click a column to sort
| Query / type | Search | Intent | Query class | Routing format | Cluster | Freq ↓ | Weight |
|---|---|---|---|---|---|---|---|
AWS SageMaker training jobs troubleshooting out of memory GPU memory PyTorch official documentation OBSERVED | Google ↗ AI Mode ↗ | — | — | — | — | 1x of 1 | |
PyTorch CUDA memory management memory_snapshot empty_cache official docs OBSERVED | Google ↗ AI Mode ↗ | — | — | — | — | 1x of 1 | |
Hugging Face Transformers performance gradient checkpointing evaluation memory official docs OBSERVED | Google ↗ AI Mode ↗ | — | — | — | — | 1x of 1 |