torchwrench.utils.rng module¶

torchwrench.utils.rng.set_seed(seed: int, cuda: bool = True, mps: bool = True, cudnn: bool = False) → None[source]¶

Set seed for random, numpy and torch packages. Optionally for CUDA, MPS and CUDNN backend.