| import numpy as np | |
| import torch | |
| import random | |
| def fixseed(seed): | |
| random.seed(seed) | |
| np.random.seed(seed) | |
| torch.manual_seed(seed) | |
| SEED = 10 | |
| EVALSEED = 0 | |
| # Provoc warning: not fully functionnal yet | |
| # torch.set_deterministic(True) | |
| torch.backends.cudnn.benchmark = False | |
| fixseed(SEED) | |