ABSA-PyTorch icon indicating copy to clipboard operation
ABSA-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon songyouwei

关于在GPU运行环境下运行

Open xu-2019 opened this issue 4 years ago • 2 comments

你好，上个月我在Google colab 中运行GPU都是可以的，但是现在运行要报错：RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor，在网上查了将pack_padded_sequence改成了pack_padded_sequence(batch, lengths = seq_lens.cpu())，也还是出现同样的错误，请问应该怎么改呢

xu-2019 avatar Dec 22 '20 04:12 xu-2019

试试改成 pack_padded_sequence(batch, lengths = seq_lens.cpu().int())

songyouwei avatar Dec 23 '20 06:12 songyouwei

谢谢您，这样成功运行了

xu-2019 avatar Dec 28 '20 03:12 xu-2019