pytorch搭建cnn报错:RuntimeError: size mismatch, m1: [10 x 43264], m2: [10816 x 2] at C...
具体报错信息:
Traceback (most recent call last):File "E:/Program Files/PyCharm 2019.2/machinelearning/homework/CNN.py", line 172, in <module>train()File "E:/Program Files/PyCharm 2019.2/machinelearning/homework/CNN.py", line 147, in trainoutput = model(data)[0]File "F:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 550, in __call__result = self.forward(*input, **kwargs)File "E:/Program Files/PyCharm 2019.2/machinelearning/homework/CNN.py", line 131, in forwardoutput = self.output(temp)File "F:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 550, in __call__result = self.forward(*input, **kwargs)File "F:\Anaconda3\lib\site-packages\torch\nn\modules\linear.py", line 87, in forwardreturn F.linear(input, self.weight, self.bias)File "F:\Anaconda3\lib\site-packages\torch\nn\functional.py", line 1610, in linearret = torch.addmm(bias, input, weight.t())
RuntimeError: size mismatch, m1: [10 x 43264], m2: [10816 x 2] at C:/w/b/windows/pytorch/aten/src\THC/generic/THCTensorMathBlas.cu:283
原因:搭建cnn时最后一层卷积然后池化之后的输出大小与最后的全连接层输出大小不匹配。
解决办法:
def forward(self, x):x = self.conv1(x)x = self.conv2(x)x = self.conv3(x)print(x.size())temp = x.view(x.shape[0], -1)output = self.output(temp)return output, x
在最后一层卷积之后,输出x的size,然后在全连接层中做出相应更正。
pytorch搭建cnn报错:RuntimeError: size mismatch, m1: [10 x 43264], m2: [10816 x 2] at C...相关推荐
- 做项目遇到问题 2 AWS NLP 剽窃RuntimeError: size mismatch, m1: [10 x 3], m2: [2 x 10]检测部署报错
报错 RuntimeError: size mismatch, m1: [10 x 3], m2: [2 x 10] 原因: train.csv 为100x4 4列 第一列 标签是否剽窃 ...
- RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at ..\aten\src\TH/generic/THTensorMath.cpp
RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at -\aten\src\TH/generic/THTensorMath.cpp: ...
- RuntimeError: size mismatch, m1: [512 x 12800], m2: [2048 x 1024] at C
RuntimeError: size mismatch, m1: [512 x 12800], m2: [2048 x 1024] at C 说一下这错误,意思就是m1和m2两个数组尺寸不一样. 为啥 ...
- pytorch RuntimeError: size mismatch, m1: [64 x 784], m2: [784 x 10] at
from torch import nnclass Mnist_Logistic(nn.Module):def __init__(self):super().__init__()self.lin=nn ...
- PyTorch RuntimeError: size mismatch, m1:
在查看torch的FastRCNNPredictor官方实现时,想弄清楚一些细节,其中nn.Linear使用的时候需要给定(in_channels, num_classes). 随便打一点测试代码,报 ...
- Pytorch-RuntimeError: size mismatch m1: [a x b], m2: [c x d]
Pytorch-RuntimeError: size mismatch m1: [a x b], m2: [c x d] 在进行深度学习的代码中,在采用线性层进行转化的时候可能出现size misma ...
- 【error】RuntimeError: size mismatch与全连接fc层
今天跑代码的时候遇到了这个错误: RuntimeError: size mismatch, m1:[1152 x 1] ,m2:[576 x 192] ,at /opt/conda/conda-bld ...
- 基于PyTorch搭建CNN实现视频动作分类任务代码详解
数据及具体讲解来源: 基于PyTorch搭建CNN实现视频动作分类任务 import torch import torch.nn as nn import torchvision.transforms ...
- RuntimeError: size mismatch
RuntimeError: size mismatch出现于pytorch框架下加载VGG11预训练模型时出现的错误.笔者初期认为,出现该错误的主要原因是输出和输入的维度不匹配.带着疑惑,我们通过输出 ...
最新文章
- 国内较强的NLP高校实验室有哪些?
- 优胜队伍跑多快?优胜秘笈是什么?直播告诉你
- 2013年第四季度总结报告
- Python必学内容:格式化输出的三种方式
- 人工智能化发展已经到了哪一步?
- 腾讯技术直播间 | 当感性遇上理性,当魔术遇上数学
- 单独运行shell脚本与crontab运行shell脚本的区别
- 【原创】指针和下标的10条对比
- python元胞自动机模拟交通_结构专栏 | 解析DEFORM软件中的元胞自动机法
- React之组件小析
- 官网opencv练习题 最简单的多物体分离技术
- demo是什么?demo有什么用?
- 2021 CSDN查看自己关注的人
- vim字符串替换命令
- 5分钟商学院-个人篇-高效能人士的思维习惯
- 抖音小店开通及保证金详情;抖音小店商品上架详细流程。丨国仁网络资讯
- 测试网速_怎样测试两台电脑之间的网速
- QT美化使用字体图标
- excel转word_pdf转word软件永久jpg图片ppt转换excel编辑器压缩合并拆分去水印,加入自律圈即可获得!!!...
- PAT 乙级 1100 校庆