DocEE icon indicating copy to clipboard operation
DocEE copied to clipboard

将"use_mention_lstm" 改为True报错

Open BudBudding opened this issue 1 year ago • 1 comments

您好,我将"use_mention_lstm" 改为True后,出现了维度报错。 当"use_mention_lstm" 为False,trigger_aware.py中 ex_idx_list = doc_batch_dict["ex_idx"] #[2532, 992, 3053, 3108, 2902, 4000, 270, 2013], 此时,doc_token_emb_list[batch_idx],# [5, 128, 768]

当"use_mention_lstm" 改为True后, ex_idx_list = doc_batch_dict["ex_idx"] #[2523, 1829, 3851, 2892, 4017, 696, 3946, 1672], 此时,doc_token_emb_list[batch_idx],# [4, 128, 768]

出现了不一致的情况,请问如何正确的使用use_mention_lstm?希望得到您的回复

BudBudding avatar Dec 05 '24 02:12 BudBudding

嗨,感谢对本项目的关注~ 请问是会报什么错误,方便贴一下完整的log吗?

Spico197 avatar Dec 05 '24 08:12 Spico197