def manual_seed(seed): r"""Sets the seed for generating random numbers. Returns a `torch._C.Generator` object. NOTE: WE REMOVE MANUAL RESEEDING ALL THE GPUS. At this point, it is not necessary; and there is not logic/reason to do it since we want only to reseed the current device. Args: seed (int): The desired seed. """ return default_generator.manual_seed(int(seed))
def manual_seed(seed): r"""Sets the seed for generating random numbers. And returns a `torch._C.Generator` object. Args: seed (int or long): The desired seed. """ import torch.cuda if not torch.cuda._in_bad_fork: torch.cuda.manual_seed_all(seed) return default_generator.manual_seed(seed)
def manual_seed(seed): r"""Sets the seed for generating random numbers. Returns a `torch.Generator` object. Args: seed (int): The desired seed. """ seed = int(seed) import torch.cuda if not torch.cuda._is_in_bad_fork(): torch.cuda.manual_seed_all(seed) return default_generator.manual_seed(seed)
def manual_seed(seed) -> torch._C.Generator: r"""Sets the seed for generating random numbers. Returns a `torch.Generator` object. Args: seed (int): The desired seed. Value must be within the inclusive range `[-0x8000_0000_0000_0000, 0xffff_ffff_ffff_ffff]`. Otherwise, a RuntimeError is raised. Negative inputs are remapped to positive values with the formula `0xffff_ffff_ffff_ffff + seed`. """ seed = int(seed) import torch.cuda if not torch.cuda._is_in_bad_fork(): torch.cuda.manual_seed_all(seed) return default_generator.manual_seed(seed)