This commit is contained in:
Eren Golge 2018-05-25 04:28:40 -07:00
parent 6e9360c455
commit a25ba81296
1 changed files with 1 additions and 1 deletions

View File

@ -200,7 +200,7 @@ class Decoder(nn.Module):
def __init__(self, in_features, memory_dim, r): def __init__(self, in_features, memory_dim, r):
super(Decoder, self).__init__() super(Decoder, self).__init__()
self.mode = mode self.r = r
self.max_decoder_steps = 200 self.max_decoder_steps = 200
self.memory_dim = memory_dim self.memory_dim = memory_dim
# memory -> |Prenet| -> processed_memory # memory -> |Prenet| -> processed_memory