Seed the pseudorandom number generator, for repeatability. Args: seed (int): random seed
(seed)
| 3 | |
| 4 | |
| 5 | def seed_experiment(seed): |
| 6 | """Seed the pseudorandom number generator, for repeatability. |
| 7 | |
| 8 | Args: |
| 9 | seed (int): random seed |
| 10 | """ |
| 11 | np.random.seed(seed) |
| 12 | torch.manual_seed(seed) |
| 13 | torch.cuda.manual_seed(seed) |
| 14 | torch.backends.cudnn.benchmark = True |
| 15 | |
| 16 | |
| 17 | def to_device(tensors, device): |