python设置随机数种子(numpy,pytorch,random)
为了保证代码能够复现,需要固定所有可能的随机数
import torch
import numpy as np
import randomdef 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.