pytorch 设置随机种子，用于测试，可复现

最新推荐文章于 2022-05-06 16:20:09 发布

MingJianGuang

最新推荐文章于 2022-05-06 16:20:09 发布

阅读量648

点赞数 1

分类专栏： pytorch 文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/qq_30636613/article/details/114277363

版权

pytorch 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

1. 核心代码

def setup_seed(seed):
    th.manual_seed(seed)
    th.cuda.manual_seed_all(seed)
    np.random.seed(seed)
    random.seed(seed)
    th.backends.cudnn.deterministic = True

setup_seed(20)

2.测试代码

import torch as th
from torch import nn
import numpy as np
import random

th.set_printoptions(precision=5)  // 设置精度

def setup_seed(seed):
    th.manual_seed(seed)
    th.cuda.manual_seed_all(seed)
    np.random.seed(seed)
    random.seed(seed)
    th.backends.cudnn.deterministic = True

setup_seed(20)

net = th.nn.Sequential(
                 nn.Linear(5,3)
                )


data = th.randn([2,3,5])
print(data)
print(net(data))

多次运行结果一致