-
Notifications
You must be signed in to change notification settings - Fork 21
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
Out of memory when saving binary weights #11
Comments
The issue still persists. Could you please guide me on how to solve it. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi,
Thank you for providing this amazing code. When I ran my code on simple network, the training code crashes at
binop.encode_rows(weight, bin_weight)
and I get this errortorch.FatalError: cuda runtime error (2) : out of memory at /opt/conda/conda-bld/pytorch_1518244421288/work/torch/lib/THC/generic/THCStorage.cu:58
I am using the same pytorch version to be consistent with your work. Could you please guide me on how to resolve this issue.
Thank you
The text was updated successfully, but these errors were encountered: