-
Notifications
You must be signed in to change notification settings - Fork 24
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
CPU Memory #5
Comments
How much batch size did you set? |
16 |
16 is not big. But you can try to decrease it, or input image size. I don't remember if I met with this issue or not. But I'm using Python3.5 and mxnet-cu80. Which line of trainunet.py cause this crash? |
I am also using Python3.5 on ubuntu14.04, not GPU. I can't remember which line. I'll let you know tomorrow. |
ncnn can run on android. |
line 73,like ‘curr_loss = F.mean(loss).asscalar()’. ncnn can run on android, but trainunet ? |
Seems your GPU's memory is not enough. You can try use small filter numbers. trainunet is only for training. Inference don't need these python scripts, only models. |
when run python trainunet.py.
File "/usr/local/lib/python3.4/dist-packages/mxnet/base.py", line 251, in check_call
raise MXNetError(py_str(_LIB.MXGetLastError()))
mxnet.base.MXNetError: [01:35:45] src/storage/./cpu_device_storage.h:73: Failed to allocate CPU Memory
The text was updated successfully, but these errors were encountered: