qq_30636613的博客

1. 定义一个网络 class MyLinear(nn.Module): def __init__(self,in_dim,out_dim): super().__init__() self.lin = nn.Linear(in_dim, out_dim) self.conv = nn.Conv2d(5,3,2) def forward(self, x): return self.lin(x) 2.生成一些数据 d

pytorch 设置随机种子，用于测试，可复现 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 t...