(seed)
| 133 | |
| 134 | |
| 135 | def seed_everything(seed): |
| 136 | torch.backends.cudnn.deterministic = True |
| 137 | torch.backends.cudnn.benchmark = False |
| 138 | torch.manual_seed(seed) |
| 139 | torch.cuda.manual_seed_all(seed) |
| 140 | np.random.seed(seed) |
| 141 | random.seed(seed) |
| 142 | |
| 143 | |
| 144 | def bool2str(b): |