训练好的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)将将池化层的张量展成一维的命令，网络就跑通了。
这个是pytorch包里面自带的文件，之前使用用vgg=models.vgg19(pretrained='True')，再进行微调和训练也没有报错，但是这次确突然报错了，不知道是什么原因。记录一下踩坑之路，踩坑经验++（smile~）

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(51249 and 250884096)相关推荐

RuntimeError: mat1 and mat2 shapes cannot be multiplied (32x7 and 784x1024) 报错解决
RuntimeError: mat1 and mat2 shapes cannot be multiplied (32x7 and 784x1024) 运行pytorch时,出现以下报错: 错误代码: ...
Pytorch vgg16 实现CIFAR10数据集分类以及RuntimeError: mat1 and mat2 shapes cannot be multiplied终极详解
最近从tensorflow转战pytorch,今天又遇到了了一个大坑:RuntimeError: mat1 and mat2 shapes cannot be multiplied,网上的结果大都模模 ...
RuntimeError: mat1 and mat2 shapes cannot be multiplied (1024x1 and 1024x3)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (1024x1 and 1024x3) 前言:在学习pytorch 搭建神经网络的时候, ...
RuntimeError: mat1 and mat2 shapes cannot be multiplied (3584x7 and 25088x4096)
RuntimeError: mat1 and mat2 shapes cannot be multiplied (3584x7 and 25088x4096) 使用VGG19提取图像特征时出现该问题 ...
RuntimeError: mat1 and mat2 shapes cannot be multiplied
RuntimeError: mat1 and mat2 shapes cannot be multiplied RuntimeError: mat1 and mat2 shapes cannot be ...
解决：RuntimeError: mat1 and mat2 shapes cannot be multiplied (8x256 and 8x256)维度不匹配问题
在设计网络是,前面几层是去噪网络,后边几层是分类网络,因为之前没有接触过分类任务,对全连接层输入维度不太理解,出现错误RuntimeError: mat1 and mat2 shapes cannot ...
RuntimeError mat1 and mat2 shapes cannot be multiplied
详细显示如下 x = self.fc(x) File "D:\Python36\lib\site-packages\torch\nn\modules\module.py", lin ...
pytorch报错：RuntimeError: mat1 and mat2 shapes cannot be multiplied (64x2500 and 3020x1600)
相信同学们在刚接触深度学习的时候老会遇到类似的问题: 这个其实就是卷积层到全连接层之间的参数数量不对应的问题卷积层的的输出结果是64*2500的矩阵,那么全连接层的第一个第一层的就应该是:2500* ...
RuntimeError: mat1 and mat2 shapes cannot be multiplied (5760x6 and 128x4)
在使用pytorch框架定义子类网络结构时,有时可能会出现mat1和mat2的形状不匹配的这种问题.如下,定义了一个7层的cnn网络: class CNN(nn.Module):def __init_ ...

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(51249 and 250884096)

问题背景

解决方法

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(51249 and 250884096)相关推荐

最新文章

热门文章

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(512*49 and 25088*4096)

问题背景

解决方法

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(512*49 and 25088*4096)相关推荐

最新文章

热门文章

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(51249 and 250884096)

训练好的vgg报错RuntimeError：mat1 and mat2 shapes cannot be multiplied(51249 and 250884096)相关推荐