andresprados / SPIGA

SPIGA: Shape Preserving Facial Landmarks with Graph Attention Networks.
https://bmvc2022.mpi-inf.mpg.de/155/
BSD 3-Clause "New" or "Revised" License
304 stars 32 forks source link

Add support for MPS on MacOS with ARM chips #29

Open walkoncross opened 1 month ago

walkoncross commented 1 month ago

Support MPS on MacOS with ARM chips.

Also allow CPU fallback when GPU and MPS are not available.