yahoo / open_nsfw

Not Suitable for Work (NSFW) classification using deep neural network Caffe models.
BSD 2-Clause "Simplified" License
5.86k stars 1.05k forks source link

performance cpu vs gpu #85

Closed lxknvlk closed 5 years ago

lxknvlk commented 5 years ago

I have tried this model with cpu-caffe, got 700ms processing time. Tried with gpu-caffe on a TeslaV100 SICK GPU RIG and got 2700ms. What am i missing? I expected at most 300ms processing time.

lxknvlk commented 5 years ago

I have fixed my confiuration, now it runs on GPU and i get about 350ms processing time. Still to long as it seems.