This commit is contained in:
Eren Golge committed 2018-05-25 04:28:40 -07:00
1 parent 9d3cda9b60
commit ff245a16cb
1 file changed
+1 -1
+1 -1
View File
@@ -200,7 +200,7 @@ class Decoder(nn.Module):
def __init__(self, in_features, memory_dim, r):
super(Decoder, self).__init__()
self.mode = mode
self.r = r
self.max_decoder_steps = 200
self.memory_dim = memory_dim
# memory -> |Prenet| -> processed_memory