2021
07-26
07-26
pytorch通过训练结果的复现设置随机种子
通过设置全局随机种子使得每次的训练结果相同可以复现defseed_torch(seed=2018):random.seed(seed)os.environ['PYTHONHASHSEED']=str(seed)np.random.seed(seed)torch.manual_seed(seed)torch.cuda.manual_seed(seed)torch.backends.cudnn.deterministic=True这里我主要讲一下模型在复现结果遇到的一些问题。首先在固定随机种子之后,每次模型训练出来的结果一定是一样的,如果不相同,那么说明训练...
继续阅读 >