Sorry to bother you again, may I ask in which part of the code did you modify the ablation experiment and the removal of the word-level graph component?
I just comment out this part, as shown below, but it doesn't seem right

Looking forword to your reply.