- 博客(1)
- 收藏
- 关注
原创 RuntimeError ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda0 Long tensor问题解决方法
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
2021-05-14 17:06:39 1461
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人