pytorch 编程总结

1. pytorch 设定随机种子

为了让实验复现，需要设定随机种子。pytorch 需要设定以下五条语句：

torch.manual_seed(myseed)			       # CPU
torch.cuda.manual_seed(myseed) 			# GPU
torch.cuda.manual_seed_all(myseed) 		   # multi-GPU
torch.backends.cudnn.deterministic = True       # cpu/gpu结果保持一致