copying a param of torch.Size([128]) from checkpoint, where the shape is torch.Size([1, 128])解决方式

最新推荐文章于 2024-08-01 10:16:18 发布

kongkongqixi

最新推荐文章于 2024-08-01 10:16:18 发布

阅读量9.2k

点赞数 2

分类专栏： python torch 机器学习算法

本文链接：https://blog.csdn.net/kongkongqixi/article/details/88182900

版权

python 同时被 3 个专栏收录

6 篇文章 0 订阅

订阅专栏

机器学习

6 篇文章 0 订阅

订阅专栏

算法

5 篇文章 0 订阅

订阅专栏

关于torch的在加载bn_inception、x_inception时出现copying a param of torch.Size([128]) from checkpoint, where the shape is torch.Size([1, 128])类型的错误，观察发现可以用过修改加载模型中权重维度解决问题。

例如：bn_inception中都是两维元组的数据转为一维：如（1,128）转为（128）可使用一下程序解决，类似问题可通过微调以下代码解决：

import torch
state_dict = torch.load('~.torch/models/bn_inception-9f5701afb96c8044.pth')
for name, weights in state_dict.items():
    # print(name, weights.size())  可以查看模型中的模型名字和权重维度
    if len(weights.size()) == 2: 判断需要修改维度的条件
        state_dict[name] = weights.squeeze(0)  去掉维度0，把(1,128)转为(128)
       # print(name,weights.squeeze(0).size()) 查看转化后的模型名字和权重维度
torch.save(state_dict, '~.torch/models/bn_inception-9f5701afb96c8044.pth')