add custom device support for RetinaFace class in detection

xinntao / facexlib

FaceXlib aims at providing ready-to-use face-related functions based on current STOA open-source methods.

MIT License

810 stars 143 forks source link

Closed caock closed 1 year ago

caock commented 2 years ago

Some code in the system will apply for resources on cuda:0, although I have specified other gpuid. This commit solves the problem