Log In Sign Up
Models:
Alyssa Salazar
AlyssaS/
MOVE
0
Downloads: 1
Data: Tabular Time Series Specialty: Endocrinology Laboratory: Blood Tests EHR: Demographics Diagnoses Medications Omics: Genomics Multi-omics Transcriptomics Wearable: Activity Clinical Purpose: Treatment Response Assessment Task: Biomarker Discovery
[c23b31]: / src / move / core / seed.py

Download this file

25 lines (19 with data), 506 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
__all__ = ["set_global_seed"]



import os

import random

from typing import Optional



import numpy as np

import torch





def set_global_seed(seed: Optional[int]) -> None:

    """Seed the random number generators.



    Args:

        seed: Seed for global random state.

    """

    if seed is None:

        return

    random.seed(seed)  #

    np.random.seed(seed)  #

    os.environ["PYTHONASSEED"] = str(seed)

    torch.manual_seed(seed)  #

    torch.cuda.manual_seed(seed)

    torch.cuda.manual_seed_all(seed)