RuntimeError: invalid argument 5: k not in range for dimension at /opt/conda/conda-bld/pytorch_15956

最新推荐文章于 2021-12-05 17:47:28 发布
最新推荐文章于 2021-12-05 17:47:28 发布
阅读量829 收藏
点赞数 1
分类专栏： 头疼的报错 文章标签： pytorch 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_46400740/article/details/120074689
版权

模型预测 topk 概率计算 类别不足 人工智能
关键词由CSDN通过智能技术生成

报错

在这里插入图片描述

解决：

包含的目标类别 少于5类 因此无法返回 top5的 概率

修改 将 topk=(1,5) 改为 （1，2）

 acc1, acc5, _ = model.get_accuracy(accuracy, output, target,topk=(1, 2))

解决！！！
在这里插入图片描述

Nova_新星
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
Pytorch one hot多标签学习报错 RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1556
Hello Word!
06-18 4347
报错： Traceback (most recent call last): File "/home/user1/main.py", line 544, in <module> main() File "/home/user1/main.py", line 273, in main validate(test_loader_lfwa, model, criterion) File "/home/user1/main.py", line 476, in validate
RuntimeError: invalid argument 5: k not in range for dimension at...
江前云后的专栏
05-24 4062
问题根源 这种情况往往出现在使用pytorch训练模型过程中，调用topk方法的时候。问题根源是，传入的参数超过了该dimension的限制。 例如tmp的shape是[4,128,1],而调用了tmp.topk(129,dim=1),显然129>128是不合理的。 另一种可能是，dataloader在load数据的时候，最后一个batch只剩了一个1个数据，而由于某些squeeze的操作导致传回的结果d的shape不是[1,128,1],而是[1,128]或者[128]。这种情况下，出现了和上面相同
RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... 错误解决方法
qq_36258740的博客
10-24 6194
RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... 这个问题是我在使用pytorch训练模型时遇到的，意思是5这个参数是无效的，我一开始认为是我数据集的问题，应该不是图片通道的问题，因为我用的都是三通道的，应该是图片尺寸不统一，就想着把它们都改成同样大小。在我改着的时候我突然意识...
RuntimeError: invalid argument 5: k not in range for dimension
u012796629的博客
10-08 382
用delf跑某个比赛的数据，热身数据只有3类，报错如题，所以导致算top5出错，因为类别数小于5，强制把topk改成top1就行
PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/g
flyfish
03-05 2669
PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/generic/THCTensorTopK.cu:26 flyfish 具体提示如下 Processinggender.py:212: UserWarning: This overload of cuda is deprecated: cuda(torch.device device, bool async,
runtime library [libssl.so.1.1] in /usr/lib/x86_64-linux-gnu may be hidden by files in:/home/
01-09
runtime library [libssl.so.1.1] in /usr/lib/x86_64-linux-gnu may be hidden by files in: /home/rw/anaconda3/lib 首先查看路径，可以看到返回结果中，第一个查找路径是anaconda的。 export $PATH bash: export...
Python RuntimeError: thread.__init__() not called解决方法
12-24
RuntimeError: thread.__init__() not called 复制代码 代码如下: class NotifyTread(threading.Thread):   def __init__(self, params):   threading.Thread.__init__(self) #here。。。。。。   self....
A multi-strategy collaborative prediction model for the runtime of online tasks in computing cluster/grid
02-21
This is usually needed to assist scheduling policies and fine tune scheduling decisions, and also used for future planning of resource allocation when conducting advance reservation. However, the ...
langchain-chatchat在window上使用cpu运行Qwen-1-8B-Chat时遇到ERROR: object
最新发布
04-26
langchain-chatchat在window上使用cpu运行Qwen-1_8B-Chat时遇到ERROR: object of type ‘NoneType‘ has no len()，这个问题其实是因为"addmm_impl_cpu_" not implemented for 'Half’这个根本原因导致的，也就是cpu...
服务器跑代码的时候，报错RuntimeError: cuda runtime error (3) :initialization error at /opt/conda/...
weixin_40286308的博客
12-05 3093
完整报错： RuntimeError: cuda runtime error (3) : initialization error at /opt/conda/conda-bld/pytorch_1579022027550/work/aten/src/THC/THCGeneral.cpp:50 THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1579022027550/work/aten/src/THC/THCGeneral.cpp line=50 e
Scrapy安装和测试小问题
江前云后的专栏
12-10 3038
安装指南 如果想通过pip安装，Windows需要先安装pip，然后在命令行中： pip install Scrapy 一般来讲会直接安装成功，但是安装过程中还是遇到了一些小bug： 1、Failed building wheel for lxml(或twisted)， 尝试用pip安装lxml时出现，结果提示如上，提示没有安装libxml12和libxslt,那就先手动去安装l
pytorch RuntimeError: size mismatch, m1: [16 x 86016], m2: [25088 x 512] at /opt/conda/conda-bld/pyt
Hello Word!
06-18 3989
报错内容： Traceback (most recent call last): File "/home/user1/main_arc_face.py", line 534, in <module> main() File "/home/user1/main_arc_face.py", line 315, in main val_loss, prec1 = validate(val_loader, model, criterion) File "/home/user1
pytorch计算top_k分类准确率
PanYHHH的博客
05-18 5410
记录一下计算top_k准确率的方法，直接调用即可: import torch def accuracy(output, target, topk=(1, 5)): maxk = max(topk) batch_size = target.size(0) _, pred = output.topk(maxk, 1, True, True) pred = pred.t() correct = pred.eq(target.view(1, -1).expand_as
Pytorch BUG 持续填坑
ZHOU的博客
04-19 5675
BUG1在做多分类的时候，计算LOSS 报错：cuda runtime error (59) : device-side assert triggered at /opt/conda/conda-bld/pytorch_1513368888240/work/torch/lib/THC/generated/../THCReduceAll.cuh:33这是因为 在计算loss的时候 data labe...
RuntimeError: NCCL error in: /opt/conda/conda-bld/pytorch_1556653215914/work/torch/lib/c10d/ProcessG
Rlin_by的博客
06-11 5687
pytorch dist 分布式训练会遇见的问题 RuntimeError: NCCL error in: /opt/conda/conda-bld/pytorch_1556653215914/work/torch/lib/c10d/ProcessGroupNCCL.cpp:272, unhandled system error
pytorch使用时遇到的坑（一）
热门推荐
永远的小白虾的博客
04-25 1万+
遇到了一个这样的错误，terminate called after throwing an instance of ‘c10::Error’ terminate called after throwing an instance of 'c10::Error' what(): CUDA error: device-side assert triggered (record at /pytor...
/opt/conda/conda-bld/pytorch_1639180588308/work/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:111
08-12
- *2* [RuntimeError: multi-target not supported at /opt/conda/conda-bld/pytorch_1549635019666/work/aten/src](https://blog.csdn.net/weixin_42419002/article/details/100988897)[target="_blank" data-...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值