Set random seed.
(seed, deterministic=False)
| 165 | |
| 166 | |
| 167 | def set_random_seed(seed, deterministic=False): |
| 168 | """Set random seed.""" |
| 169 | random.seed(seed) |
| 170 | np.random.seed(seed) |
| 171 | torch.manual_seed(seed) |
| 172 | if torch.cuda.is_available(): |
| 173 | torch.cuda.manual_seed(seed) |
| 174 | torch.cuda.manual_seed_all(seed) |
| 175 | |
| 176 | if deterministic: |
| 177 | torch.backends.cudnn.enabled = True |
| 178 | torch.backends.cudnn.benchmark = False |
| 179 | torch.backends.cudnn.deterministic = True |
| 180 | else: |
| 181 | torch.backends.cudnn.enabled = True |
| 182 | torch.backends.cudnn.benchmark = True |
| 183 | |
| 184 | |
| 185 | def get_world_size(): |
no outgoing calls
no test coverage detected