pytorch神经网络，解决输入图像大小不匹配问题 size mismatch

最新推荐文章于 2024-05-19 19:02:56 发布

xiangyong58

最新推荐文章于 2024-05-19 19:02:56 发布

阅读量1.2w

点赞数 5

分类专栏： Machine & Deep Learning 文章标签： pytorch 神经网络

本文链接：https://blog.csdn.net/xiangyong58/article/details/103822960

版权

Machine & Deep Learning 专栏收录该内容

75 篇文章 132 订阅 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

在使用PyTorch构建神经网络时遇到RuntimeError: size mismatch错误，问题源于输入图像尺寸与模型参数不一致。解决方案是检查并调整模型层的设置，确保输入与层之间的维度匹配。

摘要由CSDN通过智能技术生成

问题如下：RuntimeError: size mismatch, m1: [4 x 512], m2: [64 x 128]……

RuntimeError: size mismatch, m1: [4 x 512], m2: [64 x 128] at /opt/conda/conda-bld/pytorch_1573049304260/work/aten/src/THC/generic/THCTensorMathBlas.cu:290

RuntimeError: size mismatch, m1: [4 x 512], m2: [256 x 128] at /opt/conda/conda-bld/pytorch_1573049304260/work/aten/src/THC/generic/THCTensorMathBlas.cu:290

1. 查看网络参数：利用Pytorch搭建模型的参数,直接看程序

import torch
# 引入torch.nn并指定别名
import torch.nn as nn
import torch.nn.functional as F

class Net(nn.Module):
    def __init__(self):
        # nn.Module子类的函数必须在构造函数中执行父类的构造函数
        super(Net, self).__init__()

了解本专栏