- 博客(2)
- 收藏
- 关注
RSS订阅转载 pytorch 各个预训练网络全连接层输入参数查询
ResNet, Inception: input_size = model.fc.in_features VGG: input_size = model.classifier[0].in_features DenseNet: input_size = model.classifier.in_features SqueezeNet: input_size = model.classifier[1]....
2020-04-18 14:07:27
664
转载 pytorch 训练模型结构及参数保存
保存网络结构及其参数 torch.save(model,‘model.pth’) # 保存 model = torch.load(“model.pth”) # 加载 只加载模型参数，网络结构从代码中创建 torch.save(model.state_dict(),“model.pth”) # 保存参数 model = model() # 代码中创建网络结构 params = torch.load(...
2020-04-18 11:55:07
1380
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人