Support for Apple Metal (MPS) backend
Archie3d opened this issue · 1 comments
Archie3d commented
Please add support for the MPS backend as you do for cuda:
if torch.backends.mps.is_available():
mps = torch.device("mps")
model = model_utils.DataParallel(model)
model.to(mps)
# ... and so on...
x = x.to(mps)reinerterig commented
Has anyone looked into implementing MPS for this yet?