- 博客(2)
- 收藏
- 关注
转载 pytorch 各个预训练网络全连接层输入参数查询
ResNet, Inception: input_size = model.fc.in_featuresVGG: input_size = model.classifier[0].in_featuresDenseNet: input_size = model.classifier.in_featuresSqueezeNet: input_size = model.classifier[1]....
2020-04-18 14:07:27
623
转载 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
1331
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人