【解决】pytorch：Expected more than 1 value per channel when training, got input size [1,8]

最新推荐文章于 2023-12-12 11:59:42 发布
最新推荐文章于 2023-12-12 11:59:42 发布
阅读量422 收藏
点赞数
分类专栏： Pytorch
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/doraemon_zzn/article/details/109394016
版权

问题分析

设置的batch_size=2，但是样本数除以batch_size是无法整除的。由于整个网络使用的BatchNorm操作都是需要多于一个数据计算平均值，可能是这里造成了错误。

解决方法

找到网络的 DataLoader 部分，将 drop_last 设置为 True。
在这里插入图片描述

我叫林克不叫塞尔达
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
FCOS.Pytorch:pytorch1.x中的fcos实现
05-01
FCOS：完全卷积一阶段对象检测 由pytorch1.0实现 更新 点击率在注册 吉欧损失 点击率采样 去做 归一化回归目标 要求 的OpenCVPython的 火炬> = 1.2 火炬视觉> = 0.4。 锚点 假设白色框是gt框，不同颜色的点表示在应用ctr采样时不同要素图层的采样点。 结果 在2 1080Ti上进行训练，每个GPU 3 imgs，init lr = 1e-5余弦衰减到1e-6，但是在VOC07test上性能不佳。 在进行中央采样时，也许应该移开中心头。 预先训练的体重 由于计算资源的限制，我无法在COCO数据集上完全训练模型。 我已经将转换为我自己的。 转换后的权重可用，密码：rpni 预处理的正式实现（像素未标准化为0-1，输入img跟随BGR fomat）与我的稍有不同。 其他 基于此仓库的一些出色的工作： FCOS-Pytorch-37.2AP FCOS_
pytorch-lasso:使用PyTorch的L1正则化最小二乘
04-19
PyTorch套索 用于L1正则化最小二乘（套索）问题的PyTorch库。 该库正在进行中。 欢迎和赞赏的贡献！ 作者：Reuben Feinman（纽约大学） 乍看上去： import torch from lasso . linear import dict_learning , sparse_encode # dummy data matrix data = torch . randn ( 100 , 10 ) # Dictionary Learning dictionary , losses = dict_learning ( data , n_components = 50 , alpha = 0.5 , algorithm = 'ista' ) # Sparse Coding (lasso solve) coeffs = sparse_encode ( data , di
讲解Expected more than 1 value per channel when training, got input size torch.Size
抓饭不吃皮牙子
12-12 1164
在训练深度学习模型时，遇到错误消息"Expected more than 1 value per channel when training, got input size torch.Size"时，我们需要检查数据预处理的过程，确保输入数据的形状满足模型的要求。通过检查数据形状、数据预处理代码和模型的输入层，我们可以找出错误的原因并进行修复。这样，我们就可以成功训练模型并获得预期的结果。
BUG-Expected more than 1 value per channel when training, got input size torch.Size([1, xxx, 1, 1])
zwhdldz的博客
12-08 521
当size=Size([1, xxx, 1, 1]), size_prods是size[0] * size[2] * size[3]当size=Size([1, xxx]), size_prods是size[0]1.优先建议,调整输入尺寸或者网络结构使得 size[2]和size[3]不为1。上面的size是Size([1, xxx, 1, 1])2.方法1如果不能用,则batch_size改成大于1。就可以,这就是为什么网上通用解释都是这个.因此,为了避免这个错误,我们把。
测试时模型时出现：Expected more than 1 value per channel when training, got input size torch.Size([])问题
YancyBill的博客
07-25 262
在测试单个实验数据时报错Expected more than 1 value per channel when training, got input size torch.Size([])，猜测应该是BN层出现了问题。并且通过model.eval()后发现没有解决该问题。
python报错系列(7)--ValueError: Expected more than 1 value perchannel when training, got input size ....
热门推荐
用余生去守护的博客
01-24 1万+
活用知识和经验来寻找新点子、新创意，就是培养创造性思考所需的态度。
问题解决：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,
weixin_47725177的博客
12-27 4996
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 1280])问题解决
Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1]
It is me 的博客
07-20 1608
SSD Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1]) 原因：模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。 解决方法1 ：在torch.utils.data.DataLoader类中或自己创建的继承于DataLo..
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 128]
dotJunz的博客
02-17 2174
因为用到了BatchNorm，所以batch_size要大于1，drop_last 参数设置为True。如果是两个GPU训练，batch_size改为4。
Expected more than 1 value per channel when training, got input size torch.Size([1, 64, 1, 1])
sinat_42367115的博客
08-16 845
记录一次因BN层和Batch_size=1出现的错误和检测神经网络模型框架的有用方法
深度学习框架PyTorch：入门与实践_高清1
08-04
前言为什么写这本书2016年是属千TensorFlow的一年，凭借谷歌的大力推广，TensorFlow占据了各大媒体的头条。2017年年初，PyTorch的横空
jetson-tx2-pytorch:在Nvidia Jetson TX1TX2上安装PyTorch
05-11
在NVIDIA Jetson TX1 / TX2上安装PyTorch 是一个新的深度学习框架，可以在Jetson TX1和TX2板上很好地运行。 它安装起来相对简单快捷。 与TensorFlow不同，它不需要外部交换分区即可在TX1上构建。 尽管TX2具有足够...
early-stopping-pytorch:提前停止PyTorch
05-10
提前停止PyTorch 提早停止是一种正则化形式，用于避免过度拟合训练数据集。 提前停止跟踪验证损失，如果损失连续连续几个时期停止减少，则训练将停止。 该EarlyStopping类pytorchtool.py用于创建一个对象来跟踪验证...
Pytorch显示网络结构和参数-torchsummary
Doraemon_Zzn的博客
02-23 2007
功能：打印显示网络结构和参数 pip install torchsummary github 看一下就明白使用 import torch from torchvision import models from torchsummary import summary device = torch.device('cuda' if torch.cuda.is_available() else 'c...
Pytorch设置随机种子【结果能复现】
Doraemon_Zzn的博客
06-22 1438
最近训练只加了如下，发现每次的结果有点不一样。 torch.manual_seed(1234) 随即看了一下这方面的资料。 一般来说需要设置如下： 1.CuDNN 在CuDNN后端运行时，必须设置另外两个选项： torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = FalseCopy 警告 影响精度在小数点后几位，可以不修改，确定性模式可能会对性能产生影响，具体取决于您的型号。 2. Pytorch torc
Pytorch卷积和反卷积计算方法
Doraemon_Zzn的博客
08-11 740
torch.nn.Conv2d def __init__(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros'): Parameters in_channels (int) – Number of channels in the input i
Pytorch实现CIFAR-10数据集
Doraemon_Zzn的博客
02-28 701
练习Pytorch，做个记录 import torch import torchvision from torchvision import transforms from torch.utils.data.dataloader import DataLoader from torchsummary import summary device=torch.device('cuda') ba...
Pytorch训练自定义数据集
Doraemon_Zzn的博客
03-26 432
有点乱，先放在这里，主要做个记录。 Pytorch训练自定义数据集特征提取VGG13AlexNet整体参考预测结果 特征提取 参考了两种网络结构（一开始用的是VGG，以为用的网络有问题，又加了一个） VGG13 VGG13网络，对最后的全连接层改小了一点。 import torch.nn as nn class MyVGG13(nn.Module): def __init__(self,...
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 27, 1, 1])
最新发布
01-30
这个错误通常出现在使用深度学习... raise ValueError("Expected more than 1 value per channel when training") # 继续进行训练的代码... ``` 请注意，这只是一个示例代码，你需要根据你的具体情况进行相应的修改。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值