#通过以下代码设置随机种子和cuda相关设置,从而保证每次的训练实验结果相同
random.seed(seed)
np.random.seed(seed)
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.enabled = False
pytorch深度学习训练实验可重复设置
于 2023-07-16 11:53:21 首次发布