deleted a line the broke GravesAttention

This commit is contained in:
Adam Froghyar 2021-05-10 15:42:59 +02:00
parent f7582107da
commit 7ddc885f37
2 changed files with 0 additions and 2 deletions

View File

@ -463,7 +463,6 @@ class Decoder(nn.Module):
stop_tokens = []
t = 0
self._init_states(inputs)
self.attention.init_win_idx()
self.attention.init_states(inputs)
while True:
if t > 0:

View File

@ -375,7 +375,6 @@ class Decoder(nn.Module):
else:
self._init_states(inputs, mask=None, keep_states=True)
self.attention.init_win_idx()
self.attention.init_states(inputs)
outputs, stop_tokens, alignments, t = [], [], [], 0
while True: