(seed)
| 426 | |
| 427 | |
| 428 | def set_random_seed(seed): |
| 429 | torch.manual_seed(seed) |
| 430 | torch.cuda.manual_seed(seed) |
| 431 | torch.cuda.manual_seed_all(seed) |
| 432 | torch.backends.cudnn.benchmark = False |
| 433 | torch.backends.cudnn.deterministic = True |
| 434 | random.seed(seed) |
| 435 | np.random.seed(seed) |
| 436 | os.environ["PYTHONHASHSEED"] = str(seed) |