Nils Friederich

icon company Karlsruhe Institute of Technology icon location Karlsruhe, Germany icon location Doctoral Student in the field of ML/DL for biomedical image processing

Results 4 comments of Nils Friederich

training on Apple M1 chip using Metal Performance Shaders (MPS) float64 issue

The features relevant to this project were fixed in Pytorch2.0 (like [unique2](https://github.com/pytorch/pytorch/commit/fc59664ef4caa550ed02e5ab677134cb8d8cf74e)). But there are still some float64 uses internally. I will look at this and fix it. I will...

training on Apple M1 chip using Metal Performance Shaders (MPS) float64 issue

So, the training works. However, because nms is not yet supported for MPS (see [here](https://github.com/pytorch/pytorch/issues/78929)) the validation works quite slowly, unfortunately (because nms is on CPU). However, the MPS has...

training on Apple M1 chip using Metal Performance Shaders (MPS) float64 issue

@glenn-jocher I work on the PR :) Just a few small changes e.g. to bring tensors from int64/float64 to int32/float32.

training on Apple M1 chip using Metal Performance Shaders (MPS) float64 issue

Hi @awilbourn, have you tried device='mps'? (Pay attention to the small notation) Best, Nils