InternLM / InternLM/InternEvo

[Bug] CPU Mem utilization grows with training, when Dataloader num_workers>0

Open
#415 0 comments 0 reactions 1 assignee Claimed by @yhcc View on GitHub
bug
Dominant language
Python
Stars
421
Forks
67
PR merge metrics
No merged PRs in 30d

Description

### Describe the bug

CPU memory utilization grows with training and finally cause OOM when num_workers of Dataloader greater than 0.
Especially when more datasets are used, this mem growth phenomenon becomes more obvious.

### Environment

torch 2.3.0+cu121

### Other information

_No response_

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.