【工具代码】pytorch tensorflow设置随机种子seed

通过设置随机种子可使训练结果可复现。

pytorch

import numpy as np
import random
import torch

def setup_seed(seed):
    np.random.seed(seed)
    random.seed(seed)
    torch.manual_seed(seed) #cpu
    torch.cuda.manual_seed_all(seed)  #并行gpu
    torch.backends.cudnn.deterministic = True  #cpu/gpu结果一致
    torch.backends.cudnn.benchmark = True   #训练集变化不大时使训练加速
 
 
def main():
    setup_seed(seed)

tensorflow

import numpy as np
import random
import tensorflow as tf

def setup_seed(seed):
    np.random.seed(seed)
    random.seed(seed)
    tf.set_random_seed(seed)
 
def main():
    setup_seed(seed)

遇见你之后，我的伟大抱负和一腔热血都变成黄昏与你归家照在小路上的简单愿望。