yolov5/models
Glenn Jocher 2bf34f50fd
PyTorch Hub amp.autocast() inference (#2641)
I think this should help speed up CUDA inference, as currently models may be running in FP32 inference mode on CUDA devices unnecesarily.
2021-03-28 20:23:40 +02:00
..
hub Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py PyTorch Hub amp.autocast() inference (#2641) 2021-03-28 20:23:40 +02:00
experimental.py Update detections() self.t = tuple() (#2617) 2021-03-26 13:23:11 +01:00
export.py GPU export options (#2297) 2021-03-06 12:02:10 -08:00
yolo.py Update detections() self.t = tuple() (#2617) 2021-03-26 13:23:11 +01:00
yolov5l.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5m.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5s.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5x.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00