RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor问题解决方法
在使用rnn网络的pack_padded_sequence
出现RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor问题,可使用如下解决方式:
将
nn.utils.rnn.pack_padded_sequence(embedding, seq_lengths)
修改为:
nn.utils.rnn.pack_padded_sequence(embedding, seq_lengths.cpu())
即可解决该问题