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)
遇見你之後,我的偉大抱負和一腔熱血都變成黃昏與你歸家照在小路上的簡單願望。