np.random.seed(args.seed)
torch.manual_seed(args.seed)
torch.cuda.manual_seed_all(args.seed)
###底下这两句很重要
torch.backends.cudnn.deterministic = True
torch.backends.cudnn.benchmark = False
pytorch固定模型种子
最新推荐文章于 2024-09-14 18:35:50 发布
本文介绍了如何在PyTorch中固定随机种子,确保深度学习模型训练的可复现性。通过设置Python、numpy和PyTorch的随机种子,可以使得每次运行相同的代码得到相同的结果。
摘要由CSDN通过智能技术生成