mali19064/LSTM-CRF-pytorch-faster

并行版本的解码函数有错误

Opened this issue · 1 comments

这个代码(batch并行版本)在解码的时候是有问题的,gammar_r_l的维度是[seq_len,target_size],trainsitions的维度是[target_size,target_size],这两个变量直接相加是不行的,目测应该由于代码里没有做预测,所以忽视了这部分代码的改写

我仔细看了一下,没错误吧,gammar_r_l是[tagset_size, tagset_size],trainsitions也是[tagset_size, tagset_size]