pytorch - RuntimeError : ‘lengths’ argument should be a 1D CPU int64 tensor, 但得到 1D cuda:0 长张量

Question

我正在尝试在纸空间上使用 pytorch 和 torchtext 进行文本分类。

我明白了

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

我的PyTorch版本是1.10.1+cu102

Answer 1

我昨天刚遇到这个问题，在我的例子中，rnn pad 序列希望长度在 cpu 上，所以只需将长度放在函数调用中的 CPU 中，如下所示:

packed_sequences = nn.utils.rnn.pack_padded_sequence(padded_tensor, valid_frames.to('cpu'), batch_first=True, enforce_sorted=True) 

这可能不是您正在使用的确切函数，但我认为它将适用于大多数 rnn utils 函数。