From d5b64729d14b8063634a856fd1abc53b83085de5 Mon Sep 17 00:00:00 2001 From: Rafael Valle Date: Sun, 20 May 2018 12:22:06 -0700 Subject: [PATCH] model.py: moving for better readibility --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index 7bd8170..8ea9a2c 100644 --- a/model.py +++ b/model.py @@ -351,7 +351,6 @@ class Decoder(nn.Module): attention_weights: """ - prenet_output = self.prenet(decoder_input) cell_input = torch.cat((self.decoder_hidden, self.attention_context), -1) self.attention_hidden, self.attention_cell = self.attention_rnn( cell_input, (self.attention_hidden, self.attention_cell)) @@ -364,6 +363,7 @@ class Decoder(nn.Module): attention_weights_cat, self.mask) self.attention_weights_cum += self.attention_weights + prenet_output = self.prenet(decoder_input) decoder_input = torch.cat((prenet_output, self.attention_context), -1) self.decoder_hidden, self.decoder_cell = self.decoder_rnn( decoder_input, (self.decoder_hidden, self.decoder_cell))