yolov5

mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00

History

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

hub

Fix yolov3.yaml remove list (#6655 )

2022-02-16 10:13:05 +01:00

__init__.py

add "__init__.py" file in <models> folder

2020-06-15 14:37:07 +08:00

common.py

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

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

experimental.py

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