yolov5/models
Glenn Jocher fad27c0046
Update DDP for `torch.distributed.run` with `gloo` backend (#3680)
* Update DDP for `torch.distributed.run`

* Add LOCAL_RANK

* remove opt.local_rank

* backend="gloo|nccl"

* print

* print

* debug

* debug

* os.getenv

* gloo

* gloo

* gloo

* cleanup

* fix getenv

* cleanup

* cleanup destroy

* try nccl

* return opt

* add --local_rank

* add timeout

* add init_method

* gloo

* move destroy

* move destroy

* move print(opt) under if RANK

* destroy only RANK 0

* move destroy inside train()

* restore destroy outside train()

* update print(opt)

* cleanup

* nccl

* gloo with 60 second timeout

* update namespace printing
2021-06-19 16:30:25 +02:00
..
hub Add TransformerLayer, TransformerBlock, C3TR modules (#2333) 2021-04-01 17:26:53 +02:00
__init__.py add "__init__.py" file in <models> folder 2020-06-15 14:37:07 +08:00
common.py Parameterize max_det + inference default at 1000 (#3215) 2021-05-17 22:47:44 +02:00
experimental.py Merge `develop` branch into `master` (#3518) 2021-06-08 10:22:10 +02:00
export.py Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 2021-06-19 16:30:25 +02:00
yolo.py Update export.py, yolo.py `sys.path.append()` (#3579) 2021-06-10 15:35:22 +02: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