def seed_torch(seed):
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed) # 为CPU设置随机种子
torch.cuda.manual_seed(seed) # 为当前GPU设置随机种子(只用一块GPU)
torch.cuda.manual_seed_all(seed) # 为所有GPU设置随机种子(多块GPU)
torch.backends.cudnn.deterministic = True # NVD深度训练加速算法库
number = np.random.randint(314159) # 随机数
seed_torch(seed=number)
t_data = self.transforms(t_data) # 张量
seed_torch(seed=number)
g_data = self.transforms(g_data)