[feature request] SBR on CPU?

D-X-Y / landmark-detection

Four landmark detection algorithms, implemented in PyTorch.

MIT License

925 stars 180 forks source link

Closed ModarD closed 5 years ago

ModarD commented 5 years ago

Is it possible to run SBR on CPU without CUDA? If yes, could you please give some info on how to achieve that?

I tried to replace cuda code but I am stuck with line 51 in eval.py net = net.cuda() not sure how to replace it!

ModarD commented 5 years ago

I just removed .cuda() from everywhere in eval.py! thanks

D-X-Y commented 5 years ago

Cool!