deepspeedai / deepspeedai/DeepSpeed
[BUG]Out of memory when training, and is streaming mode supported ?
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 43.1k
- Forks
- 5k
- Avg merge
- 4d 15h
- Merged PRs (30d)
- 112
Description
description
- datasets: 1.1 GB Chinese corpus, about 2 million lines
- devices
- CPU=8 GPU=1 Memory=320G Node=1 Type=A100-SXM-80GB
question
The training process was always killed for OOM(Out Of Memory), while total size of dataset is 1.1GB, and smaller than maximum (320G)
- Program is stopped in the Tokenizing process
cd DeepSpeedExamples/applications/DeepSpeed-Chat/training/step1_supervised_finetuning/training_scripts/other_language
sh run_chinese.sh
I'm wondering if all the dataset is loaded in memory, which leads to OOM.
Do the loading process support streaming mode ?
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the command in DeepSpeed-Chat/training/step1_supervised_finetuning/training_scripts/other_language/run_chinese.sh and trace the tokenizing and dataset-loading path it invokes. Reproduce the out-of-memory termination with the reported corpus, then determine whether streaming is supported or what change is needed to avoid loading the dataset into memory.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 30/100