pytorch运行错误代码:
RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor
原因:
使用gpu读取数据错误,找到出错代码如下:
解决方法: 将seq_lengths送入cpu处理
output = classifier(inputs, seq_lengths.to('cpu'))
pytorch运行错误代码:
RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor
原因:
使用gpu读取数据错误,找到出错代码如下:
解决方法: 将seq_lengths送入cpu处理
output = classifier(inputs, seq_lengths.to('cpu'))