linzehui / mRASP

165 stars 32 forks source link

微调英德数据集,执行最后的fine-tine.sh,报错KeyError:0 #33

Closed S-GeGe closed 2 years ago

S-GeGe commented 2 years ago

@linzehui @PANXiao1994 您好,微调英德数据集,执行最后的fine-tine.sh,报错KeyError:0,报错详细信息如下,麻烦帮忙看一下。 -----> checkpoint_2_500 | en2de__wmt14_head100 <----- Traceback (most recent call last): File "/data1/home/nmt/sungege/mRASP/train/scripts/rerank_utils.py", line 152, in get_hypo_and_ref(filename, hypo_file, ref_in, ref_out, r2l=r2l) File "/data1/home/nmt/sungege/mRASP/train/scripts/rerank_utils.py", line 111, in get_hypo_and_ref assert rank < len(hypo_dict[0]) KeyError: 0