search

1adrianb / face-alignment

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

Adding apple metal support #338

Closed jimbobbennett closed 1 year ago

jimbobbennett commented 1 year ago

The latest version of PyTorch support Apple Metal (running on Apple Silicon devices, or Apple Intel devices with AMD GPUs), with the device called 'mps' - https://developer.apple.com/metal/pytorch/

This change allows the device to be set to mps rather than limiting to cpu and cuda.