Skip to content
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.

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

batch size set #184

Open
ustcjasonliu opened this issue May 2, 2024 · 2 comments
Open

batch size set #184

ustcjasonliu opened this issue May 2, 2024 · 2 comments

Comments

@ustcjasonliu
Copy link

It seems that multi batches is not supported now. I set samples_per_gpu to 2, but it went wrong. Did I set the wrong parameter?

@generalchan825
Copy link

generalchan825 commented Sep 23, 2024

hi I faced the same issue. samples_per_gpu = 1, workers_per_gpu = 16 worked, but when changing samples_per_gpu = 12, it didn't work. From #150 we have the answer.

@YTEP-ZHI
Copy link
Collaborator

Hi @ustcjasonliu @generalchan825, unfortunately, it's not supported to set bs larger than 1 in the current implementation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants