训练好的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 ·
4266 阅读 ·
5 评论