RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor(代码片段

Posted 2021-09-03 ZSYL

PyTorch 使用报错

• 报错内容
• 原因分析
• 解决方案

报错内容

RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

原因分析

因为我使用的是高版本的 torch 1.7.1，这个报错是由于torch 1.5 以上BiLSTM升级导致的。

解决方案

1. 转到CPU运行代码就不报错了！
2. 将报错的位置参数lengths转化为cpu 类型:

lengths.to("cpu")

即可解决！

参考：link