为了保证代码能够复现,需要固定所有可能的随机数
import torch
import numpy as np
import random
def seed_everywhere(seed):
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed) # if you are using multi-GPU.
np.random.seed(seed) # Numpy module.
random.seed(seed) # Python random module.