On macOS, OpenNMT-Py does not have support for metal devices (mps), but pytorch does.
It's possible to manually patch OpenNMT-Py and change torch.device('cpu') references to torch.device('mps') and get metal support to work, perhaps we could send a patch upstream to enable Metal support for everyone.
Seems kind of slow though.
On macOS, OpenNMT-Py does not have support for metal devices (mps), but pytorch does.
It's possible to manually patch OpenNMT-Py and change
torch.device('cpu')references totorch.device('mps')and get metal support to work, perhaps we could send a patch upstream to enable Metal support for everyone.Seems kind of slow though.