pytorch
the Grand Tour
这个作者很懒,什么都没留下…
展开
-
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 · 640 阅读 · 0 评论 -
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 · 1352 阅读 · 0 评论