Shortcuts

Source code for catalyst.utils.seed

import random

import numpy as np
from packaging.version import parse, Version


[docs]def set_global_seed(seed: int) -> None: """Sets random seed into PyTorch, TensorFlow, Numpy and Random. Args: seed: random seed """ try: import torch except ImportError: pass else: torch.manual_seed(seed) torch.cuda.manual_seed_all(seed) try: import tensorflow as tf except ImportError: pass else: if parse(tf.__version__) >= Version("2.0.0"): tf.random.set_seed(seed) elif parse(tf.__version__) <= Version("1.13.2"): tf.set_random_seed(seed) else: tf.compat.v1.set_random_seed(seed) random.seed(seed) np.random.seed(seed)
__all__ = ["set_global_seed"]