Too many dataloader workers #4061

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Closed

hertz-pj opened this issue May 1, 2025 · 1 comment

hertz-pj commented May 1, 2025

Too many dataloader workers: 32 (max is dataset.num_shards=1). Stopping 31 dataloader workers.

没有找到dataset.num_shards在哪里设置，请问支持多线程数据处理吗？

Collaborator

Jintao-Huang commented May 1, 2025

streaming模式嘛这个是受到iterabledataset的限制，具体可以查看datasets的文档

我们会使用dataloader_dispather，通过rank0往其他rank进行数据分发

hertz-pj closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment