Add files via upload

YoungXiyuan · Oct 15, 2019 · 0e793a6 · 0e793a6
1 parent a0c3c21
commit 0e793a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mulrel_ranker.py b/mulrel_ranker.py
@@ -292,7 +292,7 @@ def forward(self, token_ids, tok_mask, entity_ids, entity_mask, p_e_m, mtype, et
             # input()
 
             # Typing Feature
-            inputs = torch.cat([sim[indx].view(n_cands, -1),
+            inputs = torch.cat([sim[indx].view(n_cands, -1) if not basemodel == 'deeped' else local_ent_scores[indx].view(n_cands, -1),
                                 #local_ent_scores[indx].view(n_cands, -1),
                                 torch.log(p_e_m[indx] + 1e-20).view(n_cands, -1), ent_coherence.view(n_cands, -1),
                                 tm[indx].view(n_cands, -1),