-
Notifications
You must be signed in to change notification settings - Fork 82
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
Increasing batch size #82
Comments
hi @knightron0, if a batch size of 256 maxes out a 32GB V100, then a 40GB A100 should be similar. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
I'm trying to run pretraining with Resnet50 with my data, and running into out-of-memory issues with this.
Initially, I was using two V100s (32 GB) and the maximum batch size I could go to was 256. However, I can't go higher with even larger memory GPUs — I tried using an A100 both 40GB and 80GB, and the maximum batch size I could use without running into out-of-memory issues was still 256.
I'm a bit confused and was wondering if there's a knowledge gap in my understanding; let me know if I'm missing anything!
The text was updated successfully, but these errors were encountered: