Hi @MhLiao, do you have the cuda code for the prior_box_layer?
The prior_box_layer code here is a cpu version. It may take some time for copying the data from GPU to CPU, and then copying to GPU for the next layer. It may be save much time by using the prior_box_layer code of a GPU version .
Hi @MhLiao, do you have the cuda code for the prior_box_layer? The prior_box_layer code here is a cpu version. It may take some time for copying the data from GPU to CPU, and then copying to GPU for the next layer. It may be save much time by using the prior_box_layer code of a GPU version .
Thank you!