bug fix

2026-09-10 11:50:20 +08:00 · 2018-05-25 04:28:40 -07:00
parent 9d3cda9b60
commit ff245a16cb
1 changed files with 1 additions and 1 deletions
@@ -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