From 0e793a6f536af3404d038803f2159b7c53bc0c63 Mon Sep 17 00:00:00 2001 From: YoungXiyuan Date: Tue, 15 Oct 2019 15:11:48 +0800 Subject: [PATCH] Add files via upload --- mulrel_ranker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mulrel_ranker.py b/mulrel_ranker.py index 102d476..46f357a 100644 --- 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),