Loss Layer Speed #4
Description
Hi,
I am currently trying to use the loss layer with a GPU implementation of Caffe. Does the layer support GPU acceleration or are all calculations taking place on the CPU?
Thanks.
{{ message }}
Hi,
I am currently trying to use the loss layer with a GPU implementation of Caffe. Does the layer support GPU acceleration or are all calculations taking place on the CPU?
Thanks.