zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
yolov5/models
History
Glenn Jocher eb1217f3ac
Add PyTorch AMP check (#7917) 
* Add PyTorch AMP check

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Cleanup

* Cleanup

* Cleanup

* Robust for DDP

* Fixes

* Add amp enabled boolean to check_train_batch_size

* Simplify

* space to prefix

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
 		2022-05-22 13:41:18 +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 Add PyTorch AMP check (#7917) 2022-05-22 13:41:18 +02:00
experimental.py New TensorFlow `TFCrossConv()` module (#7827) 2022-05-16 17:02:16 +02:00
tf.py Add `DWConvTranspose2d()` module (#7881) 2022-05-20 16:13:40 +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