报错记录(nlp)

报错记录(nlp

1. ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

解决:Torch版本问题,Pytorch1.5以上版本升级了Bi-LSTM导致,直接的方法可以直接修改库代码(如果没有读懂当前代码可以一试)

修改的库文件:D:\software\anaconda3\envs\nlp\Lib\site-packages\torch\nn\utils\rnn.py

其中的函数:pack_padded_sequence

修改的代码部分:data, batch_sizes = _VF._pack_padded_sequence(input, lengths.cpu(), batch_first)中的lengths.cpu(),原代码:lengths

# 修改前
data, batch_sizes = \
    _VF._pack_padded_sequence(input, lengths, batch_first)
# 修改后
data, batch_sizes = \
    _VF._pack_padded_sequence(input, lengths.cpu(), batch_first)

更好的方法自然是修改自身代码。

2. RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper__index_select)

解决:依旧是Torch版本问题,可以看出高版本的Torch会给出警告,new_tensor()函数已经不推荐用于构建新张量。

UserWarning: To copy construct from a tensor, it is recommended to use sourceTensor.clone().detach() or sourceTensor.clone().detach().requires_grad_(True), rather than tensor.new_tensor(sourceTensor).
  sequences_lengths.new_tensor(torch.arange(0, len(sequences_lengths)))

基于new_tensor()构建的张量设备与原张量不一致,因此可以进一步指定予以解决。或者依据sourceTensor.clone().detach()进行创建。

# 修改前
idx_range =\
    sequences_lengths.new_tensor(torch.arange(0, len(sequences_lengths)))
# 修改后
idx_range =\
    sequences_lengths.new_tensor(torch.arange(0, len(sequences_lengths)))
idx_range = idx_range.to(sequences_lengths.device)
  • 2
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值