mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
yolov5/models
History
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
..
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) 2022-05-24 13:34:32 +02:00
tf.py YOLOv5 Apple Metal Performance Shader (MPS) support (#7878) 2022-05-24 13:34:32 +02:00
yolo.py Add `DWConvTranspose2d()` module (#7881) 2022-05-20 16:13:40 +02:00
yolov5l.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5m.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5n.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5s.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00
yolov5x.yaml YOLOv5 release v6.0 (#5141) 2021-10-11 23:47:18 -07:00