yolov5

履歴

Glenn Jocher b5de52c4cd torch.cuda.amp bug fix (#2750 ) PR https://github.com/ultralytics/yolov5/pull/2725 introduced a very specific bug that only affects multi-GPU trainings. Apparently the cause was using the torch.cuda.amp decorator in the autoShape forward method. I've implemented amp more traditionally in this PR, and the bug is resolved.		3年前
..
hub	Add TransformerLayer, TransformerBlock, C3TR modules (#2333)	3年前
__init__.py	add "__init__.py" file in <models> folder	4年前
common.py	torch.cuda.amp bug fix (#2750)	3年前
experimental.py	Update detections() self.t = tuple() (#2617)	3年前
export.py	GPU export options (#2297)	3年前
yolo.py	Add TransformerLayer, TransformerBlock, C3TR modules (#2333)	3年前
yolov5l.yaml	YOLOv5 v4.0 Release (#1837)	3年前
yolov5m.yaml	YOLOv5 v4.0 Release (#1837)	3年前
yolov5s.yaml	YOLOv5 v4.0 Release (#1837)	3年前
yolov5x.yaml	YOLOv5 v4.0 Release (#1837)	3年前