Set Random Seed

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

Guess you like

Origin www.cnblogs.com/JohnRain/p/11356806.html