(seed)
| 36 | |
| 37 | |
| 38 | def set_seed(seed): |
| 39 | torch.manual_seed(seed + comm.get_rank()) |
| 40 | os.environ['PYTHONHASHSEED'] = str(seed) |
| 41 | np.random.seed(seed) |
| 42 | torch.manual_seed(seed) |
| 43 | if torch.cuda.is_available(): |
| 44 | torch.cuda.manual_seed(seed) |
| 45 | torch.cuda.manual_seed_all(seed) |
| 46 | torch.backends.cudnn.deterministic = True |
| 47 | torch.backends.cudnn.benchmark = False |
| 48 | |
| 49 | return |
| 50 | |
| 51 | |
| 52 | def build_solver(cfg, logger): |