You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
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 ~
The text was updated successfully, but these errors were encountered: