(seed)
| 22 | # ====================================================================== |
| 23 | |
| 24 | def set_random_seed(seed): |
| 25 | np.random.seed(seed) |
| 26 | random.seed(seed) |
| 27 | torch.manual_seed(seed) |
| 28 | if torch.cuda.is_available(): |
| 29 | torch.cuda.manual_seed_all(seed) |
| 30 | # torch.backends.cudnn.determinstic = True |
| 31 | |
| 32 | |
| 33 |
no outgoing calls
no test coverage detected