RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

[WZY009]

能请问一下为什么跑自己的训练集就会出现这种问题吗

[WZY009]

如果将model/rnn.py line 83的cpu()给去掉，则会出现： RuntimeError: 'lengths' argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

[MagiaSN]

可以提供具体的命令和配置吗？