| import numpy as np | |
| import torch | |
| import random | |
| def fixseed(seed): | |
| torch.backends.cudnn.benchmark = False | |
| 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) | |