美文网首页
设置随机数种子,获得可重复的结果

设置随机数种子,获得可重复的结果

作者: LabVIEW_Python | 来源:发表于2022-07-31 15:38 被阅读0次

    在训练过程中,获得可重复的结果对于debug工作来说,非常重要。

    通常在训练代码前面加入下面的代码,可以设定确定性的运行环境。

    import random, torch, os, numpy as np
    def seed_everything(seed=42):
        os.environ['PYTHONHASHSEED'] = str(seed)
        random.seed(seed)
        np.random.seed(seed)
        torch.manual_seed(seed)
        torch.cuda.manual_seed(seed)
        torch.cuda.manual_seed_all(seed)
        torch.backends.cudnn.deterministic = True
        torch.backends.cudnn.benchmark = False
    
    seed_everything() # Set random seed
    
    # Do training Here
    

    相关文章

      网友评论

          本文标题:设置随机数种子,获得可重复的结果

          本文链接:https://www.haomeiwen.com/subject/rlvkwrtx.html