Glenn Jocher 7f7bd6fbcd Set torch.cuda.manual_seed_all() for DDP (#8688) ...

* Set `torch.cuda.manual_seed_all()` for DDP

* Update general.py

* Update general.py

2022-07-23 18:24:14 +02:00

..

aws

Fix yaml.safe_load() ignore emoji errors (#5060)

2021-10-05 13:41:52 -07:00

docker

FROM nvcr.io/nvidia/pytorch:22.06-py3 (#8600)

2022-07-17 00:55:55 +02:00

flask_rest_api

Code refactor (#7923)

2022-05-22 15:24:08 +02:00

google_app_engine

Bump pip from 19.2 to 21.1 in /utils/google_app_engine (#5661)

2021-11-15 20:10:29 +01:00

loggers

Check TensorBoard logger before adding graph (#8664)

2022-07-21 17:40:53 +02:00

__init__.py

Update bytes to GB with bitshift (#6886)

2022-03-07 19:26:37 +01:00

activations.py

[pre-commit.ci] pre-commit suggestions (#7279)

2022-04-04 22:47:00 +02:00

augmentations.py

precommit: yapf (#5494)

2022-03-31 16:52:34 +02:00

autoanchor.py

Refactor collections and fstrings (#7821)

2022-05-15 16:38:26 +02:00

autobatch.py

AutoBatch checks against failed solutions (#8159)

2022-06-09 17:15:49 +02:00

benchmarks.py

Simplify benchmarks.py assertions (#8515)

2022-07-07 23:46:55 +02:00

callbacks.py

Simplify callbacks.py return (#7333)

2022-04-07 16:15:01 +02:00

dataloaders.py

Add generator and worker seed (#8602)

2022-07-23 01:25:17 +02:00

downloads.py

Pin downloads to release version (#7790)

2022-05-13 00:24:53 +02:00

general.py

Set torch.cuda.manual_seed_all() for DDP (#8688)

2022-07-23 18:24:14 +02:00

loss.py

Fix torch long to float tensor on HUB macOS (#8067)

2022-06-01 17:34:46 +02:00

metrics.py

Update metrics.py with IoU protected divides (#8550)

2022-07-12 01:58:25 +02:00

plots.py

Add @threaded decorator (#7813)

2022-05-14 16:12:08 +02:00

torch_utils.py

assert torch!=1.12.0 for DDP training (#8621)

2022-07-18 15:05:58 +02:00