(seed, cuda=True)
| 163 | self.file.close() |
| 164 | |
| 165 | def set_seed(seed, cuda=True): |
| 166 | random.seed(seed) |
| 167 | np.random.seed(seed) |
| 168 | torch.manual_seed(seed) |
| 169 | if cuda: |
| 170 | torch.backends.cudnn.deterministic = True |
| 171 | torch.cuda.manual_seed(seed) |
| 172 | |
| 173 | def set_requires_grad(nets, requires_grad=False): |
| 174 | """Set requies_grad=Fasle for all the networks to avoid unnecessary computations |