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

..

aws

Fix yaml.safe_load() ignore emoji errors (#5060)

2021-10-05 13:41:52 -07:00

docker

FROM nvcr.io/nvidia/pytorch:22.08-py3 (#9520)

2022-09-20 22:57:42 +02:00

flask_rest_api

Allow multiple-model serving from Flask REST API (#8973)

2022-08-15 23:42:25 +02:00

google_app_engine

Bump pip from 19.2 to 21.1 in /utils/google_app_engine (#5661)

2021-11-15 20:10:29 +01:00

loggers

Update Comet links (#9587)

2022-09-25 20:12:57 +02:00

segment

Add segment line predictions (#9571)

2022-09-25 14:52:49 +02:00

__init__.py

Standardize warnings with WARNING ⚠️ ... (#9467)

2022-09-18 16:15:25 +02:00

activations.py

[pre-commit.ci] pre-commit suggestions (#7279)

2022-04-04 22:47:00 +02:00

augmentations.py

Update albumentations (#9503)

2022-09-21 19:53:26 +02:00

autoanchor.py

Standardize warnings with WARNING ⚠️ ... (#9467)

2022-09-18 16:15:25 +02:00

autobatch.py

AutoBatch report include reserved+allocated (#9491)

2022-09-19 12:18:55 +02:00

callbacks.py

Daemon plot_labels() for faster start (#9057)

2022-08-21 03:47:37 +02:00

dataloaders.py

generator seed fix for DDP mAP drop (#9545)

2022-09-24 19:35:07 +02:00

downloads.py

Update default GitHub assets (#9573)

2022-09-25 02:59:25 +02:00

general.py

Apple MPS -> CPU NMS fallback strategy (#9600)

2022-09-26 14:13:03 +02:00

loss.py

Fix torch long to float tensor on HUB macOS (#8067)

2022-06-01 17:34:46 +02:00

metrics.py

Standardize warnings with WARNING ⚠️ ... (#9467)

2022-09-18 16:15:25 +02:00

plots.py

New scale_segments() function (#9570)

2022-09-24 16:02:41 +02:00

torch_utils.py

Standardize warnings with WARNING ⚠️ ... (#9467)

2022-09-18 16:15:25 +02:00

triton.py

Detect.py supports running against a Triton container (#9228)

2022-09-24 00:56:42 +02:00