RuntimeError:"length"参数应为1D CPU int64张量，但得到1D cuda

Question

我正在尝试在paperspace上使用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函数。