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