代码改变世界

随机种子设置

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