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

"detection_accuracy_layer" dosen't have gpu version #95

Open
ubenz55555 opened this issue Mar 3, 2018 · 1 comment
Open

"detection_accuracy_layer" dosen't have gpu version #95

ubenz55555 opened this issue Mar 3, 2018 · 1 comment

Comments

@ubenz55555
Copy link

ubenz55555 commented Mar 3, 2018

Hi zhaoweicai

Is your accuracy loss layer called "detection_accuracy_layer.cpp"

and your bounding box loss called "detection_loss_layer.cu" under mscnn/src/caffe/layers/ ?

I have noticed that your detection_accuracy_layer only have cpp version, which is

run by cpu only,and there is no detection_accuracy_layer.cu for gpu version ,but your

detection_loss_layer have both cpu and gpu version . Since you both use "detection_accuracy_layer.cpp"

and "detection_loss_layer.cu" to train our model, I wonder that you only use cpu to train your

model since there is no "detection_accuracy_layer.cu" . Is that right ? Thanks ~

@ubenz55555 ubenz55555 changed the title About the loss layer "detection_accuracy_layer" dosen't have gpu version Mar 3, 2018
@zhaoweicai
Copy link
Owner

Under the GPU mode, when a layer only has a CPU version, CPU will be used for its computation. But the computation for the other layers will still use GPU. Since the "detection_accuracy_layer" doesn't have too much computation, it is OK to use CPU for its computation.

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

2 participants