Set random seed. Args: seed (int): Seed to be used. deterministic (bool): Whether to set the deterministic option for CUDNN backend, i.e., set `torch.backends.cudnn.deterministic` to True and `torch.backends.cudnn.benchmark` to False. Default:
(seed, deterministic=False)
| 15 | |
| 16 | |
| 17 | def set_random_seed(seed, deterministic=False): |
| 18 | """Set random seed. |
| 19 | Args: |
| 20 | seed (int): Seed to be used. |
| 21 | deterministic (bool): Whether to set the deterministic option for |
| 22 | CUDNN backend, i.e., set `torch.backends.cudnn.deterministic` |
| 23 | to True and `torch.backends.cudnn.benchmark` to False. |
| 24 | Default: False. |
| 25 | """ |
| 26 | random.seed(seed) |
| 27 | np.random.seed(seed) |
| 28 | torch.manual_seed(seed) |
| 29 | torch.cuda.manual_seed_all(seed) |
| 30 | if deterministic: |
| 31 | torch.backends.cudnn.deterministic = True |
| 32 | torch.backends.cudnn.benchmark = False |
| 33 | |
| 34 | |
| 35 | def train_model(model, |