Glenn Jocher c215878f11 YOLOv5 Apple Metal Performance Shader (MPS) support (#7878) ...

* Apple Metal Performance Shader (MPS) device support

Following https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/

Should work with Apple M1 devices with PyTorch nightly installed with command `--device mps`. Usage examples:
```bash
python train.py --device mps
python detect.py --device mps
python val.py --device mps
```

* Update device strategy to fix MPS issue

2022-05-24 13:34:32 +02:00

..

aws

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

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

docker

Update Dockerfile-arm64 (#7860)

2022-05-17 16:35:12 +02:00

flask_rest_api

Code refactor (#7923)

2022-05-22 15:24:08 +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

Refactor collections and fstrings (#7821)

2022-05-15 16:38:26 +02:00

__init__.py

Update bytes to GB with bitshift (#6886)

2022-03-07 19:26:37 +01:00

activations.py

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

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

augmentations.py

precommit: yapf (#5494)

2022-03-31 16:52:34 +02:00

autoanchor.py

Refactor collections and fstrings (#7821)

2022-05-15 16:38:26 +02:00

autobatch.py

Add PyTorch AMP check (#7917)

2022-05-22 13:41:18 +02:00

benchmarks.py

Add check_yaml() to benchmarks.py (#7916)

2022-05-21 19:39:19 +02:00

callbacks.py

Simplify callbacks.py return (#7333)

2022-04-07 16:15:01 +02:00

dataloaders.py

cv2.resize interpolation fix (#7903)

2022-05-20 12:59:05 +02:00

downloads.py

Pin downloads to release version (#7790)

2022-05-13 00:24:53 +02:00

general.py

Windows check_file() fix (#7938)

2022-05-23 16:15:38 +02:00

loss.py

Swap unsafe_chunk() for chunk() (#7362)

2022-04-10 13:53:28 +02:00

metrics.py

Bug fix mAP0.5-0.95 (#6787)

2022-05-20 11:33:10 +02:00

plots.py

Add @threaded decorator (#7813)

2022-05-14 16:12:08 +02:00

torch_utils.py

YOLOv5 Apple Metal Performance Shader (MPS) support (#7878)

2022-05-24 13:34:32 +02:00