๋ชฉ๋ก2025/02/05 (1)
๐บ๐๐๐๐๐
[Pytorch] Fixing Random Seed
# fixing random seeddef seed_everything(seed:int = 1004): random.seed(seed) np.random.seed(seed) os.environ["PYTHONHASHSEED"] = str(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) # current gpu seed torch.cuda.manual_seed_all(seed) # All gpu seed torch.backends.cudnn.deterministic = True # type: ignore torch.backends.cudnn.benchmark = False seed = seed_everyt..
Python
2025. 2. 5. 15:25