Skip to content
This repository was archived by the owner on May 15, 2024. It is now read-only.

Commit 708a804

Browse files
committed
fix bug
1 parent ef96b22 commit 708a804

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

network.py

-2
Original file line numberDiff line numberDiff line change
@@ -433,8 +433,6 @@ def detail_forward_disentangle(self, incoming):
433433
self.beta = torch.sum(valid_wiki_h_n1 * incoming.hidden.h_n, dim = 2)
434434
self.beta = torch.t(self.beta) # [valid_num, wiki_len]
435435

436-
incoming.acc.rel.append(torch.index_select(self.beta, 0, incoming.state.reverse_valid_sen).cpu().tolist())
437-
438436
mask = torch.arange(self.beta.shape[1], device=self.beta.device).long().expand(self.beta.shape[0], self.beta.shape[1]).transpose(0, 1) # [wiki_sen_num, valid_num]
439437
expand_wiki_num = valid_wiki_num.unsqueeze(0).expand_as(mask) # [wiki_sen_num, valid_num]
440438
reverse_mask = (expand_wiki_num <= mask).float() # [wiki_sen_num, valid_num]

0 commit comments

Comments
 (0)