关于‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor报错的解决方发

复现论文的时候遇到的BUG，后来发现是torch版本升级的原因，发个博客记录一下。

解决方法是需要修改RNN.py文件中的代码。

源代码在244行上下：

    data, batch_sizes = \_VF._pack_padded_sequence(input, lengths, batch_first)

我们需要改成：

 data, batch_sizes = \_VF._pack_padded_sequence(input, lengths.cpu(), batch_first)

关于‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor报错的解决方发相关推荐

es7学习笔记 cpu负载不均衡、超长fullGC、大量400报错
现象:往es7集群中推数时,发生如下情况接口出现很多400 发现集群中某台机器cpu被怼爆发生fullGC 产生400报错的原因是es7做了熔断优化,当jvm内存使用超过阈值,为了避免丑陋的oom ...
RuntimeError: ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor
活久见 kernel:NMI watchdog: BUG: soft lockup - CPU#8 stuck for 28s! [xsoftdd/12:0]
1. 引入在一台linux机器上工作,没敲入任何命令,但命令行里突然出现 "kernel:NMI watchdog: BUG: soft lockup - CPU#8 stuck for ...
[报错] TypeError: run() argument after * must be an iterable, not int
[报错] TypeError: run() argument after * must be an iterable, not int 问题描述: 在多线程操作,调用threading.Thread ...
报错解决之 __init__() missing 1 required positional argument: ‘dtypes‘
Index 目录索引报错信息解决方法参考报错信息深度学习神经网络模型训练的时候,可能会报出如下错误: __init__() missing 1 required positional arg ...
Tensor for ‘out‘ is on CPU, Tensor for argument #1 ‘self‘ is on CPU
1.问题模型训练完后进行测试,报错 RuntimeError: Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU ...
Python报错TypeError: Tuple or struct_time argument required
错误原因,类型出错误啦,转换不了! 我的解决: >>> time.mktime(time.localtime()) 1545053905.0 具体就看下面的! 我的报错: >& ...
missing 1 required positional argument: 'on_delete'报错解决方案
最近在使用Python的Django框架开发web站点,通过models.py文件建表后,执行数据库迁移(命令行:mange.py makemigrations)时报错,下面是查看官方文档后找到的解决 ...
Oracle报错TNS-12532: TNS:invalid argument
今天一个同事遇到了一个oracle问题,在数据库本机通过sqlplus登录sys用户时报TNS-12532: TNS:invalid argument,这个错误. 造成这个错误的原因主要有三个,一般排 ...

关于‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor报错的解决方发

关于‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor报错的解决方发相关推荐

最新文章

热门文章