Set random seed.
(seed, deterministic=False)
| 32 | |
| 33 | |
| 34 | def set_random_seed(seed, deterministic=False): |
| 35 | """Set random seed.""" |
| 36 | random.seed(seed) |
| 37 | np.random.seed(seed) |
| 38 | torch.manual_seed(seed) |
| 39 | torch.cuda.manual_seed_all(seed) |
| 40 | if deterministic: |
| 41 | torch.backends.cudnn.deterministic = True |
| 42 | torch.backends.cudnn.benchmark = False |
| 43 | |
| 44 | |
| 45 | @torch.no_grad() |