search

midasklr / facedetection_android.pytorch

Light-weight face detection on Android with pytorch model
20 stars 6 forks source link

I want to know wether this inference on cpu or gpu? #1

Open YAwei666 opened 3 years ago

YAwei666 commented 3 years ago

thank you

midasklr commented 3 years ago

thank you

Today, PyTorch executes the models on the CPU backend pending availability of other hardware backends such as GPU, DSP, and NPU... see:https://pytorch.org/tutorials/recipes/mobile_perf.html?highlight=mobile