将"use_mention_lstm" 改为True报错

[BudBudding]

您好，我将"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？希望得到您的回复

[Spico197]

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