Skip to content
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

Solver.solve() interface raises error if the given blob is larger then a number. #1916

Closed
erogol opened this issue Feb 20, 2015 · 5 comments
Closed

Comments

@erogol
Copy link
Contributor

erogol commented Feb 20, 2015

solver = caffe.SGDSolver(SOLVER_PYTHON_FILE)

solver.net.set_input_arrays(np.ascontiguousarray(train_img_blob, dtype=np.float32),
np.ascontiguousarray(train_label_blob, dtype=np.float32))
solver.solve()

This is what I've done for training a network over python interface. Given train data blob is multiple of the batch size (32). This code works for train data blob size < 512, otherwise it gives this as an output

I0220 15:08:21.057770 4054 net.cpp:213] Network initialization done.
I0220 15:08:21.057777 4054 net.cpp:214] Memory required for data: 556160
I0220 15:08:21.057915 4054 solver.cpp:42] Solver scaffolding done.
Segmentation fault (core dumped)

I guess something wrong or missing from my humble side.

@rohrbach
Copy link

I believe this should go to caffe-users -- this issues tracker is primarily for Caffe development discussion. Thanks!

@erogol
Copy link
Contributor Author

erogol commented Feb 22, 2015

Yes, but I beleive so this is an development issue to track and nothing to do with the caffe-users

@shelhamer
Copy link
Member

This is actually a duplicate of #1470 -- no single blob can be larger than 2 GB given the current types.

@rohrbach
Copy link

sorry @erogol for commenting/closing this incorrectly, but @shelhamer clarified it.

@erogol
Copy link
Contributor Author

erogol commented Feb 23, 2015

Thanks for the clarification @shelhamer and np @rohrbach . Best :)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants