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

Is there any plan to realize these CPU parts? #44

Open
zzningxp opened this issue Sep 20, 2016 · 7 comments
Open

Is there any plan to realize these CPU parts? #44

zzningxp opened this issue Sep 20, 2016 · 7 comments

Comments

@zzningxp
Copy link

zzningxp commented Sep 20, 2016

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?

@zzningxp zzningxp changed the title Why SoftmaxWithLossLayer<Dtype>::Forward_cpu in src/caffe/layers/softmax_loss_layer.cpp is NOT_IMPLEMENTED? Is there any plan to realize these CPU parts? Sep 20, 2016
@zzningxp
Copy link
Author

Furthermore, I think that to test an image by using the pre-trained model on the CPU machine is really in demand.
Also, I think that there is only one CPU forward (psroi_pooling_layer.cpp) should be implemented to fulfill the CPU test.

@xiaoxiongli
Copy link

xiaoxiongli commented Sep 22, 2016

I think you'd better to buy a GPU. such as M40. it is really fast if you use a GPU.

@zzningxp
Copy link
Author

I have GPU machine, and I need to test the forward pregress on the a CPU machine.

@chenxistephen
Copy link

I have the same issue to test the CPU forward time (even I have a GPU machine).

@Uchanka
Copy link

Uchanka commented Mar 7, 2017

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.
If you still want one of these, I'll push this to my github ASAP.
@zzningxp

@buaadf
Copy link

buaadf commented Jul 13, 2017

@Grillnov I need the file now, have you push it to your gitHub, i didn't find it in your github. thank you

@liyuanyaun
Copy link

@Grillnov could you share the file? thank you very much

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

No branches or pull requests

6 participants