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