-
由 Glenn Jocher 提交于
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/9596Signed-off-by:
Glenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by:
Glenn Jocher <glenn.jocher@ultralytics.com>