- 博客(2)
- 收藏
- 关注
原创 pytorch 循环生成网络结构
由于网络通常非常深,我们一层一层写会耗费很大的精力。而且修改起来也比较麻烦,不方便维护。网络通常由一些相似的块组成,我们写出一个块,循环生成整个网络。 nn.Sequential def __init__(self, num_layers=1, activation=torch.nn.ReLU()): super(Classifier3, self).__init__() self.num_layers = num_layers self.dense1 = torch.nn.Sequ
2021-07-21 19:27:06 817
原创 pytorch 模型保存与加载
模型保存 torch.save(model, os.path.join(checkpoint_dir, f'model_{epoch}.pth')) 模型加载 checkpoint = torch.load(checkpoint_file) model=checkpoint.to(device) 只加载变量: checkpoint = torch.load(lists[epoch]) model.load_state_dict(checkpoint.state_dict()) 当模型训练使用的显卡编号
2021-07-21 18:48:00 294
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人