Glenn Jocher c4c0ee8fc3
Apple MPS -> CPU NMS fallback strategy (#9600)
Until more ops are fully supported this update will allow for seamless MPS inference (but slower MPS to CPU transfer before NMS, so slower NMS times).

Partially resolves https://github.com/ultralytics/yolov5/issues/9596

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2022-09-26 14:13:03 +02:00
..
2022-09-25 20:12:57 +02:00
2022-09-21 19:53:26 +02:00
2022-09-24 16:02:41 +02:00