pytorch设置随机种子

	SEED = 0
	torch.manual_seed(SEED)
	torch.cuda.manual_seed(SEED)

刚开始用了上面的代码,结果跑测试的时候每次每次测试结果都不一样。看博客又加了一句,改成了如下:

    SEED = 0
    torch.manual_seed(SEED)
    torch.cuda.manual_seed(SEED)
    np.random.seed(SEED)

每次跑出来的测试结果终于一样了。

参考博文:https://cloud.tencent.com/developer/article/1149041


版权声明:本文为xuan971130原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。