RuntimeError: The size of tensor a (128) must match the size of tensor b (16) at non-singleton dimen

最新推荐文章于 2024-04-23 14:42:19 发布
最新推荐文章于 2024-04-23 14:42:19 发布
阅读量6.8k 收藏 7
点赞数 2
分类专栏： 踩坑 文章标签： python 深度学习 pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_40740309/article/details/115432786
版权

RuntimeError: The size of tensor a (128) must match the size of tensor b (16) at non-singleton dimension 0

这个错误出现的原因有好多种！！
其中一个比较可能的原因是因为：封装dataloader的时候，最后剩下的不足一个batchsize！ 自带的dataloader就会有这个现象。

改正：

batch_size_s = len(targets) #不足一个batch_size直接停止训练
if batch_size_s < BATCH_SIZE:
    break

就把最后一个不足batchsize的丢掉！！

吃嘛嘛香zx
关注
  • 2
    点赞
  • 踩
  • 7
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
专栏目录
RuntimeError:The size of tensor a (128) must match the size of tensor b (256) at non-singleton dime
qq_45013475的博客
10-13 2万+
RuntimeError:The size of tensor a (128) must match the size of tensor b (256) at non-singleton dimension 3 使用Pytorch跑网络的时候发生了如标题所述的报错，好多地方有类似的情况，不同体现在最后是dimension或是dimension 1之类。 原因：下采样次数必上采样次数多一次，导致输出的图像尺寸比输入尺寸小一倍，最后在计算loss（MSE）时出现报错，只要将上采样次数增加，使得最后输出尺寸
[yolov5]The size of tensor a (3078) must match the size of tensor b (3) at non-singleton dimension报错
Satellite的博客
05-09 2871
问题描述: 在使用yolov5训练自己制作的数据集时出现报错 背景: 数据集是自己制作的yolo格式数据集,图片较大,没有做resize处理,分辨率为4000*3000,猜测可能是数据集出现问题 在网上搜索报错信息: 解决方案一： img = Image.open(image_path) 改为 img = Image.open(image_path).convert(‘RGB’)。 解决方案二： 问题1::RuntimeError: DataLoader worker (pid XXX) is kille
【目标检测】FPN特征金字塔完整流程详解
最新发布
CV在读
04-23 1225
特征金字塔介绍
大模型部署避坑指南--RuntimeError: The size of tensor a (32000) must match the size of tensor b (32001) ...
bulucc的博客
05-04 771
将vicuna-7b-delta-v0 替换为 vicuna-7b-delta-v1.1 版本。fastchat版本与模型版本 不兼容。
RuntimeError: The size of tensor a (40) must match the size of tensor b (41) at non-singleton dimens
喜碧CatBrother
10-25 2万+
RuntimeError: The size of tensor a (40) must match the size of tensor b (41) at non-singleton dimension 3 此类还是维度问题，建议尝试以下几种解决方案： 1、 model.avg_pool = nn.AvgPool2d(kernel_size=1, stride=1, padding=0) mo...
RuntimeError: The size of tensor a (32) must match the size of tensor b (28) at non-singleton
qq_38284961的博客
08-29 3万+
这个错误主要是因为，数据的数量不能整除batch size的数量，改正方法有两种 1. 将batch size改称可以被 数据量整除即可。 2. 将dataloader 中的 drop_last改称False也可。 推荐使用第二种方法更加简单。 ...
【已解决】RuntimeError: The size of tensor a (32) must match the size of tensor b (128) at non-singleton
BetrayFree的博客
11-06 2万+
上述使用sim这个矩阵和adj1这个矩阵需要进行相乘，上面报错的原因就是因为对应维度不对，所以没有办法运算，那就把对应的改一下就可以了。这个就是训练用到的维度和其中参与计算的维度是不一致的，拿我本身的来说。这里边是128，也就是和上面问题是一致的，那改为32就可以了也即。发现了这个得到的维度不对，只需要改一下对应的就可以了。
pytorch模型提示超出内存RuntimeError: CUDA out of memory.
01-20
跑模型时出现RuntimeError: CUDA out of memory.错误 查阅了许多相关内容，原因是：GPU显存内存不够 简单总结一下解决方法： 将batch_size改小。 取torch变量标量值时使用item()属性。 可以在测试阶段添加如下代码：...
RuntimeError: Cannot run the event loop while another loop is running(目前没有解决)
12-21
代码如下: import tornado.ioloop import tornado.web from tornado.httpclient import HTTPClient, AsyncHTTPClient class MainHandler(tornado.web.RequestHandler): # 同步 def get(self): ...
解决pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\\python3.6\\lib\\site-packa的问题
01-21
标题用pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\python3.6\lib\site-packages\numpy\init.py 1.file→settings→project interpreter→+（建议用pychon3.6版本，我...
pytorch dataloader 取batch_size时候出现bug的解决方式
12-20
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 281 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333 2、 RuntimeError: ...
RuntimeError:The size of tensor a (128) must match the size of tensor b (256) at non-singleton dime
m0_45924886的博客
05-10 4409
print(stat(model,input_size=[4,256,256])) #input_size是输入图片的大小。model=InpaintGenerator(1) #1是InpaintGenerator的参数。
Pytorch多标签损失函数MultiLabelSoftMarginLoss报错RuntimeError: The size of tensor a (128) must match the size
Hello Word!
06-16 7455
代码： import torch.nn as nn criterion = nn.MultiLabelSoftMarginLoss(weight=w) # train on celeba train set and lfwa train set train_dataset = CelebA(args.data, 'train_40_att_list.txt', transforms.Compose([ transforms.Resize((250, 250)), transforms.Ran
pytorch.tensor格式图像的resize操作
wzg2016的博客
10-24 2万+
在pytorch中，输入网络的图像的shape=[B,C,H,W]. 有时我们需要在网络中对图像张量进行resize操作，这时就要用到transforms.Resize([H,W]) 操作。示例如下： import cv2 import numpy as np import torch from torchvision.transforms import Resize im1 = cv2.imread("./datasets/frame_0001.png").transpose([2,0,1])
The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 1
weixin_44337238的博客
04-20 1万+
debug记录
【解决错误】The size of tensor a (8) must match the size of tensor b (64) at non-singleton dimension 1
热门推荐
学习带来的快乐最持久~
12-03 5万+
一、报错 RuntimeError: Given groups=1, weight of size 64 64 1 1, expected input[16, 48, 40, 40] to have 64 channels, but got 48 channels instead 二、原因 错误分析：这里面说的是要求输入的权重是[64，64,1，1]，其中第二个维度的要求是64，但是给定的输入却是[16, 48, 40, 40]，导致输入的第二个维度48和64不匹配。 三、解决方案 一般出错都
RuntimeError: The size of tensor a (20) must match the size of tensor b (10) at non- singleton dime
CSDN 精品推荐
08-04 2760
使用 `PyG` 搭建 `GCN` 实现链路预测任务时，出现了如下问题 `RuntimeError: The size of tensor a (20) must match the size of tensor b (10) at non- singleton dimension 2`
深度学习中训练时候遇到这些错误 RuntimeError、IndexError、ValueError如何解决
m0_63669388的博客
07-17 4338
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same CUDA error: device-side assert triggeredRuntimeError: Expected all tensors to be on the same device, The size of tensor a (5) must match the size of
DRealSR运行中遇到The size of tensor a (360) must match the size of tensor b (361) at non-singleton dimens
程序员养成之路的博客
03-09 1110
DRealSR代码运行解决。
RuntimeError: The size of tensor a (128) must match the size of tensor b (122) at non-singleton dimension 3
08-15
- *1* [RuntimeError: The size of tensor a (4) must match the size of tensor b (3) at non-singleton dimensio](https://blog.csdn.net/weixin_46135327/article/details/130805823)[target="_blank" data-...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值