yolov5/utils
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
..
aws Download COCO and VOC by default (#3608) 2021-06-14 13:58:54 +02:00
flask_rest_api Update JSON response (#3139) 2021-05-12 18:23:50 +02:00
google_app_engine Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) 2021-06-09 21:14:09 +02:00
wandb_logging Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 2021-06-19 16:30:25 +02:00
__init__.py initial commit 2020-05-29 17:04:54 -07:00
activations.py ACON Activation batch-size 1 bug patch (#2901) 2021-04-25 22:54:07 +02:00
autoanchor.py Update check_requirements() exclude list (#2974) 2021-04-29 21:16:23 +02:00
datasets.py Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 2021-06-19 16:30:25 +02:00
general.py Update `check_file()` (#3622) 2021-06-15 13:21:04 +02:00
google_utils.py Parse URL authentication (#3424) 2021-06-03 18:44:58 +02:00
loss.py Add --label-smoothing eps argument to train.py (default 0.0) (#2344) 2021-03-29 18:45:46 +02:00
metrics.py ConfusionMatrix `normalize=True` fix (#3587) 2021-06-11 11:46:05 +02:00
plots.py Ignore Seaborn plot warnings (#3576) 2021-06-10 13:51:29 +02:00
torch_utils.py Update DDP for `torch.distributed.run` with `gloo` backend (#3680) 2021-06-19 16:30:25 +02:00