- 博客(2)
- 收藏
- 关注
RSS订阅原创 训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(512*49 and 25088*4096)
问题背景将训练好的图片分类vgg模型用到新的数据集上进行图片分类的时候出现了以下问题：解决方法结合VGG的网络架构：发现池化层的输出张量为51277，对应报错的512*49，其无法与第一个全连接层FC1的权重系数相乘，继而和bias相加作为FC1的输出。但是在输出到全连接层之前，网络的forward函数中应当定义输入张量的处理过程，查看在VGG模型类的定义中：存在x = torch.flatten(x, 1)存在展成一位张量效果的命令，再加上x = x.view(-1, 512*7*7)将将
2022-05-16 21:57:30
4128
5
原创 win10+anaconda+python3.8虚拟环境中openslide包的安装全记录
win10+anaconda+python3.8虚拟环境中安装openslide包全记录
2022-04-13 16:49:13
3869
15
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人