深度学习|随机数固定

import torch
import numpy as np
import random

seed=0

random.seed(seed)
np.random.seed(seed)
if torch.cuda.is_available():
    torch.cuda.manual_seed_all(seed)

torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)

本作品采用知识共享署名 4.0 国际许可协议进行许可

深度学习|随机数固定

文章评论