代码改变世界

随机种子设置

2019-08-15 11:14  JohnRain  阅读(625)  评论(0)  编辑  收藏  举报 
np.random.seed(SEED)
torch.manual_seed(SEED)
torch.cuda.manual_seed(SEED)