Closed ducha-aiki closed 1 year ago
On mps there are some problems with mapping the final output, however it is likely problem of pytorch mps backend, which still has lots of bugs.
The cpu version works just fine on Apple device:
Looks good to me :)
Defaults are not changing. Now
demo_match.py
can run ontorch.device('mps')
device, if it such device is specified