1adrianb / face-alignment

:fire: 2D and 3D Face alignment library build using pytorch
https://www.adrianbulat.com
BSD 3-Clause "New" or "Revised" License
7.03k stars 1.34k forks source link

Parallelizing on the CPU #266

Open JOST777 opened 3 years ago

JOST777 commented 3 years ago

Is it possible or has anyone tried to parallelize face-alignment on the CPU? I tried to do so with from multiprocessing import Pool, but I ran into this pytorch issue.

I'm assuming the thing to do is to just run on a GPU, but I thought I would put it out there in case someone already knows a way to make multiprocessing work.