update decoder

model/decoder.py

@@ -65,10 +65,7 @@ class Decoder(nn.Module):
         
         output = output.squeeze(1)
         embedded = embedded.squeeze(1)
-        
         output_context = torch.cat((output, context, embedded), dim=1)
         output_context = self.layer_norm(output_context)
-        
         prediction = self.fc_out(output_context)
-        
         return prediction, decoder_hidden, decoder_cell, attention_weights