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版权协议,转载请附上原文出处链接和本声明。