-
Notifications
You must be signed in to change notification settings - Fork 136
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
How to prevent GPU OOM? #186
Comments
Yes, you probably want smaller networks (e.g. less filters, less layers) and smaller batch sizes. |
In |
Did you have a look here? |
Yes, I had a look at #174 but O can't seem to make any modifications that work. Right now I have the current repo
(It crashes...) |
From your stack trace, AlphaZero.jl crashes during the gradient-update phase, not during self-play. So my guess is that you should also lower your |
I'm not sure it makes any difference in Julia or if it should incur a crash during the gradient-update phase, but in your |
Thanks, replaced the dot with a comma (old eyes...)
|
Delete your |
I missed Update: It's still going, now on iteration 4; won 32% on iteration 3). |
I followed the readme:
This led to:
Is there some way to tweak parameters so that I can prevent running out of memory?
The text was updated successfully, but these errors were encountered: