pytorch中，设置随机种子，适用于NER实验

原创已于 2022-05-15 15:50:06 修改 · 343 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #机器学习

于 2020-12-28 16:26:08 首次发布

pytorch 专栏收录该内容

47 篇文章

订阅专栏

博客围绕Pytorch在深度学习和机器学习领域的应用展开。Pytorch是重要工具，在这两个信息技术热门领域发挥着关键作用，能帮助开发者更高效地进行模型构建与训练等操作。

部署运行你感兴趣的模型镜像

import random
import numpy as np
import torch
def setup_seed(seed):
	torch.manual_seed(seed)				#为cpu分配随机种子
	if torch.cuda.is_available():
		torch.cuda.manual_seed(seed)	#为gpu分配随机种子
		torch.cuda.manual_seed_all(seed)#若使用多块gpu，使用该命令设置随机种子
	random.seed(seed)
	np.random.seed(seed)
	torch.backends.cudnn.deterministic = True
	torch.backends.cudnn.benchmark = False

setup_seed(seed)