-
Notifications
You must be signed in to change notification settings - Fork 262
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
Added gpu_memory::buffer class to help in no-pool config #68
Conversation
I built this with Make and without CNMeM. It built fine and passed all the caffe tests. But then I tried to train AlexNet on it with DIGITS and got this error:
I'm going to rebuild with DEBUG symbols to see if I can find any more information ... |
Actually, it didn't pass all the Caffe tests. I spoke too soon.
|
Can you restart this one? I just pushed a fix. From: Luke Yeager notifications@github.com Actually, it didn't pass all the Caffe tests. I spoke too soon. [ RUN ] CuDNNConvolutionLayerTest/1.TestGradientCuDNN Reply to this email directly or view it on GitHubhttps://github.com//pull/68#issuecomment-153531256. This email message is for the sole use of the intended recipient(s) and may contain reply email and destroy all copies of the original message. |
It restarted when you pushed a commit. |
Added gpu_memory::buffer class to help in no-pool config
Using temp_buffer class helps retain memory in layers as we did before.