Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Return early in `reset_seed` (#11983)

This commit is contained in:
ananthsub 2022-02-25 15:58:27 -08:00 committed by GitHub
parent 4f68c635a1
commit 244f365fae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
pytorch_lightning/utilities/seed.py
View File

@ -87,9 +87,10 @@ def reset_seed() -> None:
If :func:`pytorch_lightning.utilities.seed.seed_everything` is unused, this function will do nothing.
"""
seed = os.environ.get("PL_GLOBAL_SEED", None)
if seed is None:
return
workers = os.environ.get("PL_SEED_WORKERS", "0")
if seed is not None:
seed_everything(int(seed), workers=bool(int(workers)))
seed_everything(int(seed), workers=bool(int(workers)))
def pl_worker_init_function(worker_id: int, rank: Optional[int] = None) -> None: # pragma: no cover