yolov5/models/hub
Ding Yiwei 1148e2ea63
Add TransformerLayer, TransformerBlock, C3TR modules (#2333)
* yolotr

* transformer block

* Remove bias in Transformer

* Remove C3T

* Remove a deprecated class

* put the 2nd LayerNorm into the 2nd residual block

* move example model to models/hub, rename to -transformer

* Add module comments and TODOs

* Remove LN in Transformer

* Add comments for Transformer

* Solve the problem of MA with DDP

* cleanup

* cleanup find_unused_parameters

* PEP8 reformat

Co-authored-by: DingYiwei <846414640@qq.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2021-04-01 17:26:53 +02:00
..
anchors.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov3-spp.yaml PANet anchor order correction (#1420) 2020-11-17 10:13:00 +01:00
yolov3-tiny.yaml Update yolov3-tiny.yaml (#1823) 2021-01-02 13:01:05 -08:00
yolov3.yaml Pycocotools best.pt after COCO train (#1616) 2020-12-06 14:58:33 +01:00
yolov5-fpn.yaml PANet anchor order correction (#1420) 2020-11-17 10:13:00 +01:00
yolov5-p2.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5-p6.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5-p7.yaml YOLOv5 v4.0 Release (#1837) 2021-01-04 19:54:09 -08:00
yolov5-panet.yaml PANet anchor order correction (#1420) 2020-11-17 10:13:00 +01:00
yolov5l6.yaml Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
yolov5m6.yaml Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
yolov5s-transformer.yaml Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 2021-04-01 17:26:53 +02:00
yolov5s6.yaml Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00
yolov5x6.yaml Update plot_study() (#2112) 2021-02-01 13:51:11 -08:00