RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/generic/THC

说晚安了吗#

已于 2024-06-18 16:44:06 修改

阅读量151

点赞数 4

文章标签：人工智能机器学习深度学习

于 2024-05-20 18:34:48 首次发布

本文链接：https://blog.csdn.net/weixin_46552088/article/details/139059882

版权

RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/generic/THCTensorTopK.cu:24

作者在训练一个分类模型时在train时没有报错，但是在验证val时报了这个错误

RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/generic/THCTensorTopK.cu:24

意思是当前只有5各分类，无法计算top5准确率
找到代码修改分类数解决

acc1, acc5 = accuracy(output, target, topk=(1, 4))

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

说晚安了吗#

关注关注

4
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

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 ...

RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... 错误解决方法

qq_36258740的博客

10-24

6316

RuntimeError:invalid argument 5:k not in range for dimension at /pytorch/ate ... 这个问题是我在使用pytorch训练模型时遇到的，意思是5这个参数是无效的，我一开始认为是我数据集的问题，应该不是图片通道的问题，因为我用的都是三通道的，应该是图片尺寸不统一，就想着把它们都改成同样大小。在我改着的时候我突然意识...

参与评论您还未登录，请先登录后发表或查看评论

'RuntimeError: invalid argument 1: A should be square at...'解决方法

兔奔的博客

10-11

1295

RuntimeError: invalid argument 1: A should be square at c:\n\pytorch_1559129895673\work\aten\src\th\generic/THTensorLapack.cpp:534 出现该错误的原因是因为矩阵不是方阵，在tensor.inverse()处抛出该异常。解决方法是检查数据是否出现所有任何两个维度维数均不相...

解决：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension

u012953338的专栏

05-23

1万+

原博客地址解决：RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 544 and 1935 in dimension 2 at ../aten/src/TH/generic/THTensor.cpp:711 这种错误有两种可能：　　1.你输入的图像数据的维度不完全是一样的，比如是训练的数据有100组，其中99组是256*256，但有一组是384*384，这样会导致Pyto

Pytorch遇到的错误及解决方法

zx93的博客

09-15

3238

Pytorch遇到的错误及解决方法

PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/g

二分掌柜的

03-05

2746

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,

Ubuntu 安装 .NET SDK / .NET Runtime 支持 Demo for Ubuntu

08-03

Ubuntu 安装 .NET SDK / .NET Runtime 支持 Demo for Ubuntu https://blog.csdn.net/loveric/article/details/132086925

C++使用onnxruntime/opencv对onnx模型进行推理(附代码)

05-23

C++使用onnxruntime/opencv对onnx模型进行推理(附代码)

Autoware.Universe 运行样例高精地图到 ~/autoware-map

最新发布

09-20

Autoware.Universe 运行样例，...gdown -O ~/autoware_map/ 'https://docs.google.com/uc?export=download&id=1499_nsbUbIeturZaDj7jhUownh5fvXHd' unzip -d ~/autoware_map ~/autoware_map/sample-map-planning.zip

手动解决Cannot find module '@sentry/webpack-plugin'等问题

10-06

解决Cannot find module '@sentry/webpack-plugin'，Cannot find module '@sentry/browser'，Cannot find module '@sentry/cli'等问题。<br/> 下载后，解压后文件夹，放在vue项目的node_modules文件夹下，即可。

NCRF++错误解决

weixin_43821843的博客

06-03

1084

NCRF++: An Open-source Neural Sequence Labeling Toolkit RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/generic/THCTensorTopK.cu:21 解决的方法是：解码配置中的参数：### Decode ...

RuntimeError: invalid argument 4: out of range at pytorch/torch/lib/TH/generic/THTensor.c:439

u011961856的专栏

12-02

4699

使用from torchvision.utils import save_image保存图像时,出现错误:File “/usr/local/lib/python2.7/dist-packages/torchvision/utils.py”, line 51, in save_imagegrid = make_grid(tensor, nrow=nrow, padding=padding) File

RuntimeError: invalid argument 5: k not in range for dimension at...

江前云后的专栏

05-24

4184

问题根源这种情况往往出现在使用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 2: view size is not compatible with input tensor‘s size and str

weixin_44313451的博客

03-09

369

错误： RuntimeError: invalid argument 2: view size is not compatible with input tensor’s size and stride…… box_input = box_input.view(batch_sizeself.nr_boxesself.nr_frames, 4) 修改为：box_input = box_input.contiguous().view(batch_sizeself.nr_boxesself.nr_frames,

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

weixin_46400740的博客

09-03

872

报错解决：包含的目标类别少于5类因此无法返回 top5的概率修改将 topk=(1,5) 改为（1，2） acc1, acc5, _ = model.get_accuracy(accuracy, output, target,topk=(1, 2)) 解决！！！

pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 1 a

热门推荐

公众号/知乎: AI算法札记，欢迎关注交流

11-28

3万+

1.pytorch报错 Namespace(batchSize=16, gpu='0,1', lr=0.001, nepoch=210, num_workers=8) Epoch: 0 Epoch:0 [0|173089] loss:3.747539 Epoch:0 [1|173089] loss:3.629151 Epoch:0 [2|173089] loss:3.731266 Traceba...

pytorch RuntimeError: invalid argument 1: .

sinat_36024346的博客

01-16

2715

当使用torch.view()时出现 RuntimeError: invalid argument 1: input is not contiguous at /pytorch/torch/lib/TH/generic/THTensor.c:231 可以使用 tensor.contiguous() 解决 mask[idx, :].view(bat_size, 1) 变成 mask[idx,...

RuntimeError: invalid argument 0: Tensors must have same number of dimensions: got 2 and 1 at ..

dfd4578575的博客

10-28

9021

使用pytorch Dataloade加载数据时遇到的错误：解决方案：当if条件不满足时列表维度数不对，改成：

RuntimeError: invalid argument 3: out of range at C:\w\1\s\windows\pytorch\aten\src\THC/generic/THCTensor.cpp:382

03-12

这个错误通常是由于在PyTorch中使用了超出范围的索引或维度大小导致的。具体来说，它表示在C:\w\1\s\windows\pytorch\aten\src\THC/generic/THCTensor.cpp文件的382行发生了错误，错误类型是RuntimeError，错误信息是invalid argument 3: out of range。要解决这个问题，你可以检查以下几个方面： 1. 确保你的索引或维度大小是有效的，不超出张量的范围。 2. 检查你的代码中是否有任何硬编码的索引或维度值，可能需要根据实际情况进行调整。 3. 确保你的输入数据的形状与你的操作相匹配，例如在进行张量操作之前，可以使用.size()方法检查张量的形状。如果你能提供更多的上下文或代码片段，我可以给出更具体的建议。