Jan Hajek 7a0a81fd1d GPU export options (#2297) ... * option for skip last layer and cuda export support * added parameter device * fix import * cleanup 1 * cleanup 2 * opt-in grid --grid will export with grid computation, default export will skip grid (same as current) * default --device cpu GPU export causes ONNX and CoreML errors. Co-authored-by: Jan Hajek <jan.hajek@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>		2021-03-06 12:02:10 -08: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	Update inference default to multi_label=False (#2252)	2021-02-19 15:39:09 -08:00
experimental.py	EMA bug fix 2 (#2330)	2021-03-01 17:13:34 -08:00
export.py	GPU export options (#2297)	2021-03-06 12:02:10 -08:00
yolo.py	Anchor override (#2350)	2021-03-02 23:08:21 -08: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