| import numpy as np | |
| import random | |
| import torch | |
| def set_seed(config): | |
| print('config.seed',config.seed) | |
| random.seed(config.seed) | |
| np.random.seed(config.seed) | |
| torch.manual_seed(config.seed) | |
| torch.cuda.manual_seed(config.seed) | |
| torch.cuda.manual_seed_all(config.seed) |