首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >运行时错误: mat1和mat2形状不能在pytorch中相乘

运行时错误: mat1和mat2形状不能在pytorch中相乘
EN

Stack Overflow用户
提问于 2021-10-30 09:00:56
回答 1查看 19.6K关注 0票数 3

我刚开始深入学习,我用下面的代码创建了一个用于预测植物病害的模型

代码语言:javascript
运行
复制
class CNN_Model(nn.Module):
  def __init__(self):
    super(CNN_Model, self).__init__()
    self.cnn_model = nn.Sequential(
        nn.Conv2d(3, 16, 3),    
        nn.ReLU(),
        nn.MaxPool2d(2, 2),     
        nn.Conv2d(16, 32, 5),   
        nn.ReLU(),
        nn.MaxPool2d(2, 2),     
    )

    self.fc_model = nn.Sequential(
        nn.Flatten(),           
        nn.Linear(800, 300),    
        nn.ReLU(),
        nn.Linear(300, 38),     
        nn.Softmax(dim=1)
    )

  def forward(self, x):
      x = self.cnn_model(x)
      x = self.fc_model(x)

      return x
代码语言:javascript
运行
复制
model = CNN_Model()

out = model(imgs)
out

当我试图运行上面的代码时,我得到的错误mat1和mat2不能相乘。我已经尝试了类似的问题的答案,但我的问题仍然没有解决。

代码语言:javascript
运行
复制
RuntimeError                              Traceback (most recent call last)
/tmp/ipykernel_66/1768380315.py in <module>
----> 1 out = model(imgs)
      2 out

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)
   1049         if not (self._backward_hooks or self._forward_hooks or self._forward_pre_hooks or _global_backward_hooks
   1050                 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1051             return forward_call(*input, **kwargs)
   1052         # Do not call functions when jit is used
   1053         full_backward_hooks, non_full_backward_hooks = [], []

/tmp/ipykernel_66/1577403502.py in forward(self, x)
     26   def forward(self, x):
     27       x = self.cnn_model(x)
---> 28       x = self.fc_model(x)
     29 
     30       return x

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)
   1049         if not (self._backward_hooks or self._forward_hooks or self._forward_pre_hooks or _global_backward_hooks
   1050                 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1051             return forward_call(*input, **kwargs)
   1052         # Do not call functions when jit is used
   1053         full_backward_hooks, non_full_backward_hooks = [], []

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/container.py in forward(self, input)
    137     def forward(self, input):
    138         for module in self:
--> 139             input = module(input)
    140         return input
    141 

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py in _call_impl(self, *input, **kwargs)
   1049         if not (self._backward_hooks or self._forward_hooks or self._forward_pre_hooks or _global_backward_hooks
   1050                 or _global_forward_hooks or _global_forward_pre_hooks):
-> 1051             return forward_call(*input, **kwargs)
   1052         # Do not call functions when jit is used
   1053         full_backward_hooks, non_full_backward_hooks = [], []

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/linear.py in forward(self, input)
     94 
     95     def forward(self, input: Tensor) -> Tensor:
---> 96         return F.linear(input, self.weight, self.bias)
     97 
     98     def extra_repr(self) -> str:

/opt/conda/lib/python3.7/site-packages/torch/nn/functional.py in linear(input, weight, bias)
   1845     if has_torch_function_variadic(input, weight):
   1846         return handle_torch_function(linear, (input, weight), input, weight, bias=bias)
-> 1847     return torch._C._nn.linear(input, weight, bias)
   1848 
   1849 

RuntimeError: mat1 and mat2 shapes cannot be multiplied (32x119072 and 800x300)

请有人帮我解决这些错误。

EN

回答 1

Stack Overflow用户

发布于 2021-10-30 09:45:21

大小不匹配误差显示为32x119072 and 800x300。第一个形状是指输入张量,第二个是层的参数。如果您查看您的模型定义,您将看到它与第一个完全连接的层相匹配,在平坦层之后的那个层。实际上,nn.Linear(800, 300)期待的是800-feature张量,但得到的是119072-feature张量。

您需要修改这个线性层以匹配传入的张量扁平的空间形状。但是请注意这个值将如何依赖于输入CNN的图像,最终这将决定输入分类器的张量的大小。解决这一问题的一般方法是使用自适应层:例如nn.AdaptiveMaxPool2d,它将始终提供相同的输出形状,而不管输入维度大小如何。

票数 8
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/69778174

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档