-
Notifications
You must be signed in to change notification settings - Fork 540
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
Is there any plan to realize these CPU parts? #44
Comments
Furthermore, I think that to test an image by using the pre-trained model on the CPU machine is really in demand. |
I think you'd better to buy a GPU. such as M40. it is really fast if you use a GPU. |
I have GPU machine, and I need to test the forward pregress on the a CPU machine. |
I have the same issue to test the CPU forward time (even I have a GPU machine). |
I've implemented the cpu compatible version of PSROIPoolingLayer from this original psroi_pooling_layer.cu, although I'm not pretty sure if it's buggy or not. |
@Grillnov I need the file now, have you push it to your gitHub, i didn't find it in your github. thank you |
@Grillnov could you share the file? thank you very much |
SoftmaxWithLossLayer::Forward_cpu in src/caffe/layers/softmax_loss_layer.cpp is NOT_IMPLEMENTED, while
./src/caffe/layers/box_annotator_ohem_layer.cpp
./src/caffe/layers/psroi_pooling_layer.cpp
./src/caffe/layers/smooth_L1_loss_layer.cpp
are the same.
Is there any plan to realize these CPU part?
The text was updated successfully, but these errors were encountered: