查看: 1483|回复: 0

pytorch数据类型报错解决方案:

[复制链接]

9

主题

47

帖子

123

积分

注册会员

Rank: 2

积分
123
发表于 2018-10-15 15:01:16 | 显示全部楼层 |阅读模式
写一个自己遇到的pytorch数据类型报错的例子之前在运行代码时遇到一个数据类型错误:其报错如下:
  1. RuntimeError: Expected object of type Variable[torch.LongTensor] but found type Variable[torch.cuda.ByteTensor] for argument #1 ‘argument1’
复制代码
这个解决办法为:
pytorch框架在存储labels时,采用LongTensor来存储,所以在一开始dataset返回label时,就要返回与LongTensor对应的数据类型,即numpy.int64
这个希望对大家有帮助


回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表