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)