gather_out_cuda(): Expected dtype int64 for index

最新推荐文章于 2023-12-06 21:06:08 发布
最新推荐文章于 2023-12-06 21:06:08 发布
阅读量3.6k 收藏 4
点赞数 3
分类专栏： 错误解决 代码复现
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/zhongruizhe/article/details/121059558
版权

PyTorch CUDA Image Captioning bug修复 类型错误
关键词由CSDN通过智能技术生成

gather_out_cuda(): Expected dtype int64 for index
在 Dual-Level Collaborative Transformer for Image Captioning复现代码的时候发现bug。
问题原因：由于pytorch版本不对
因为我的显卡是gtx3090,所以cuda版本只能是11.0，安装的pytorch1.7
解决思路：
初始代码：

selected_beam = selected_idx / candidate_logprob.shape[-1]

更改后的代码：

selected_beam = selected_idx // candidate_logprob.shape[-1]

更改路径：image-captioning-DLCT-main/models/beam_search/beam_search.py,的第118行

附件：开源代码路径
感谢：https://giters.com/kdexd/virtex/issues/20提供修改思路！！！

zhongruizhe
关注
  • 3
    点赞
  • 踩
  • 4
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
专栏目录
stanfordnlp使用报错“index_select(): Expected dtype int32 or int64 for index”解决方法
AdrenalinmyVein的博客
05-21 3696
stanfordnlp使用报错（float index相关）解决方法
Gather_Town_Gamepad:用游戏控制器玩 Gather Town！
08-03
Gather_Town_Gamepad 用游戏控制器玩 Gather Town！ 笔记 支持Xbox Game Controller, PS DUALSHOCK 4. (其他应该也支持) 可以同时检测和支持多个游戏控制器。 关键地图 游戏控制器按钮 电脑键盘 左摇杆 WASD X ...
解决RuntimeError: gather_out_cuda(): Expected dtype int64 for index
lingyunyuny的博客
07-08 1059
原因应该是和pytorch版本有关系，根据报错，找到原因为target.unsqueeze(2)的dtype是int32类型的，把它转为int64了，可以正常运行。在训练模型过程中有如下语句，结果出现报错。
动手学强化学习-模仿学习RuntimeError: gather(): Expected dtype int64 for index
最新发布
oceancoco的博客
12-06 705
当出现RuntimeError这个错误信息的时候大概率是因为索引类型不对，追根溯源找到应该的索引类型，再找到错误源进行更改即可。即actions的类型转换为int64 经过输出发现原来的actions为int32类型的。经过不断查资料，发现gather只能索引int64类型的数据，所以将原来的。：原网站上的代码有一点问题，报错一下信息。
torch.gather()的理解
qq_3311753064的博客
03-20 3677
在看蘑菇书的时候出现了这样的代码： '''计算当前(s_t,a)对应的Q(s_t, a)''' '''torch.gather:对于a=torch.Tensor([[1,2],[3,4]]),那么a.gather(1,torch.Tensor([[0],[1]]))=torch.Tensor([[1],[3]])''' q_values = self.policy_net(state_batch).gather(dim=1, index=action_batch) # 等
RuntimeError: scatter(): Expected dtype int64 for index
weixin_45928096的博客
01-08 1万+
RuntimeError: scatter(): Expected dtype int64 for index 1.报错原因： scatter要求数据是int64类型，而我在定义tensor时写的是torch.Tensor(x)，应该写成torch.LongTensor(x)，指定为int64类型。 2.解决方法 找到原数据的定义方式，改！ 一般在dtype=np.int64；dtype=np.float32中 （多数定义函数都有dtype属性） 最好int和float的位数要一致
DBMS_STATS.GATHER_TABLE_STATS详解.pdf
02-01
作用:DBMS_STATS.GATHER_TABLE_STATS统计表,列,索引的统计信息. DBMS_STATS.GATHER_TABLE_STATS的语法如下: DBMS_STATS.GATHER_TABLE_STATS ( ownname VARCHAR2, tabname VARCHAR2, partname VARCHAR2, estimate_...
picture_gather_and_transport5.rar_Linux 图片
09-21
在Linux中，这通常通过V4L2（Video for Linux Two）接口来实现。V4L2提供了一套API，使得应用程序能够访问各种视频输入和输出设备，包括内置或外接的摄像头。通过编程调用这些API，可以捕获实时图像，甚至进行视频流...
metasploit_gather_exchange:Exchange Server的Metasploit开发后收集模块
03-12
Exchange Server的Metasploit开发后收集模块 一个用于从Exchange Server收集信息的Metasploit框架模块。 该模块通过Meterpreter会话从运行在受感染目标上的Exchange Server收集信息。 它使用“ Exchange命令行管理...
sgbuf.rar_SCATTER GATHER
09-24
Scatter-Gather缓冲区驱动是Linux内核中的一个重要概念，主要用于优化I/O操作，特别是针对块设备和网络传输。在传统的I/O操作中，数据通常从一个单一的内存区域读取或写入到另一个单一的区域，而在Scatter-Gather...
pytorch系列（一）常见错误
qq_30049011的博客
01-26 1万+
pytorch使用（一）常见错误 1、RuntimeError: scatter_(): Expected dtype int64 for index. RuntimeError: scatter_(): Expected dtype int64 for index. scatter要求数据是int64类型，而我在定义tensor时写的是torch.Tensor(x)，应该写成torch.LongTensor(x)，指定为int64类型。其他数据类型的定义方法可参考： https://pytorch.or
Tensorflow 实践中常见的错误解决方法
Kaiyin
07-26 1万+
1、ValueError: setting an array element with a sequence. 这个错误通常是由于train_x,train_y 和 mask（test_x,test_y,mask）出现了问题。这个时候可以尝试打印部分train_x,部分train_y或mask。尤其注意他们的长度信息是否对应。遇到过的出错有： X_train, X_test, y_train,...
PyTorch编写代码遇到的问题
flyfish
03-03 1万+
训练人脸检测mtcnn遇到的问题 flyfish no module named xxx xxx为自定义文件夹的名字 因为搜索不到，所以将当前路径加入到包的搜索目录 解决方法： import sys sys.path.append(’…’) #将上层目录加入到搜索路径中 sys.path.append(’/home/xxx’) # 绝对路径 import os sys.path.append(os...
pytorch使用笔记
weixin_47895059的博客
07-25 804
1.torch.max() torch.max(input,0/1) 其中input是一个tensor，返回两个tensor，如果第二个参数是零，返回每行最大值和所在列数，反之对应 a = torch.tensor([[1,5,62,54], [2,6,2,6], [2,65,2,6]]) x=torch.max(a,1) y=torch.max(a,0) print(x) print(y) ...
Pytorch高版本(1.0)跑程序出现的几个错误
yo跟着新宇走
04-09 1万+
Pytorch高版本1.0跑程序出现的几个错误错误 output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]错误 NotADirectoryError: [WinError 267] 目录名称无效。错误 Expected object of scalar type Long but got scalar ...
一个numpy int64的坑
热门推荐
weixin_33769125的博客
09-26 2万+
2019独角兽企业重金招聘Python工程师标准>>> ...
RuntimeError: gather_out_cuda(): Expected dtype int64 for index
05-02
这个错误通常是由于在 PyTorch 的 gather 函数中使用了错误的索引数据类型引起的。在 gather 函数中，索引必须是整数类型，通常是 int64。建议检查您的代码并确保传递给 gather 函数的索引是整数类型。如果您不确定，可以将索引强制转换为 int64，如下所示： ``` index = index.to(torch.int64) output = torch.gather(input, dim, index) ``` 如果问题仍然存在，请检查索引张量的形状是否与输入张量匹配，并且索引是否超出了输入张量的范围。

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

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

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值