model.load_state_dict(state_dict, strict=False)

我们在构造好了一个模型后，可能要加载一些训练好的模型参数。举例子如下：

假设 trained.pth 是一个训练好的网络的模型参数存储

model = Net()是我们刚刚生成的一个新模型，我们希望model将trained.pth中的参数加载加载进来，但是model中多了一些trained.pth中不存在的参数，如果使用下面的命令：

state_dict = torch.load('trained.pth')
model.load_state_dict(state_dict)

会报错，说key对应不上，因为model你强人所难，我堂堂trained.pth没有你的那些个零碎玩意，你非要向我索取，我上哪给你弄去。

但是model不干，说既然你不能完全满足我的需要，那么你有什么我就拿什么吧，怎么办呢？下面的指令代码就行了。

model.load_state_dict(state_dict, strict=False)

model.load_state_dict(state_dict, strict=False)相关推荐

strict=False 但还是size mismatch for []: copying a param with shape [] from checkpoint,the shape in cur
strict=False 但还是size mismatch for []: copying a param with shape [] from checkpoint,the shape in cur ...
model.parameters()，model.state_dict()，model .load_state_dict()以及torch.load()
一.model.parameters()与model.state_dict() model.parameters()与model.state_dict()都是Pytorch中用于查看网络参数的方法一 ...
PyTorch加载模型model.load_state_dict()问题，Unexpected key(s) in state_dict: “module.features..，Expected .
希望将训练好的模型加载到新的网络上.如上面题目所描述的,PyTorch在加载之前保存的模型参数的时候,遇到了问题. Unexpected key(s) in state_dict: "mod ...
pytorch训练的pt模型转换为onnx（nn.DataParallel()、model、model.state_dict()）
pt转onnx流程与常见问题 pt转onnx流程 pt转onnx流程 1.读取pt模型文件,文件既可以是torch.save(model,path)整体保存的模型,也可以是保存的字典文件. // An ...
CoTNet 修改笔记
CoTNet 依赖cupy 关于cupy的文章: cupy_jacke121的专栏-CSDN博客_cupy 调用测试代码: import torchfrom models import cotnet5 ...
Convolutional-MLPs改造
Convolutional-MLPs 人脸识别用,112*112返回7*7特征: 调用代码: import torchfrom src import convmlp_sif __name__ == ' ...
目标检测模型---SSD
SSD 1.SSD网络模型 2. 先验框的准备(default box//Prior box) 3. VOC格式数据集的准备 4. 模型训练 4.1 模型如何从特征中获取预测结果 4.2 ground ...
【小样本分割 2020 ICCV】PANet
文章目录 [小样本分割 2020 ICCV]PANet 1. 简介 2. 网络 2.1 整体架构 2.2 原型学习 2.3 非参数度量学习 2.4 原型对齐正则化 3. 代码 3.1 backbone ...
Pytorch单机多卡加速
忙了两个月从收到原始数据到最后在工程项目中加载成功完成测试,好像从元旦上班后就再没休息过,昨天项目通过三期评审终于可以喘口气补点作业了.(年前写的文章,今天转过来了) 多卡并行一定要使用 torch ...
基于YOLOV5动态检测19种类别
#1 makeTxt.py(将训练数据自动划分为训练集.验证集和测试集)... 2 #2 voc_label.py(将VOC格式数据集转换成yolo数据集)... 3 #3 tube.yaml 6 # ...

model.load_state_dict(state_dict, strict=False)

model.load_state_dict(state_dict, strict=False)相关推荐

最新文章

热门文章