Rename field

Author: emalgorithm

models/lstm_decoder.py

@@ -9,7 +9,7 @@ def __init__(self, hidden_size, output_size, device, attention=False, pointer_ne
         self.hidden_size = hidden_size
         self.output_size = output_size
         self.embedding = nn.Embedding(output_size, hidden_size).to(device)
-        self.gru = nn.LSTM(hidden_size, hidden_size).to(device)
+        self.lstm = nn.LSTM(hidden_size, hidden_size).to(device)
         self.out = nn.Linear(hidden_size, output_size).to(device)
         self.softmax = nn.LogSoftmax(dim=1)
         self.attention = attention
@@ -45,7 +45,7 @@ def forward(self, input, hidden, encoder_hiddens, input_seq=None):
             output = self.attention_combine(output)
 
         output = F.relu(output)
-        output, hidden = self.gru(output, hidden)
+        output, hidden = self.lstm(output, hidden)
         output = self.softmax(self.out(output[0]))
 
         return output, hidden

models/lstm_encoder.py

@@ -8,14 +8,14 @@ def __init__(self, input_size, hidden_size, device):
         self.hidden_size = hidden_size
 
         self.embedding = nn.Embedding(input_size, hidden_size).to(device)
-        self.gru = nn.LSTM(hidden_size, hidden_size, batch_first=True).to(device)
+        self.lstm = nn.LSTM(hidden_size, hidden_size, batch_first=True).to(device)
         self.device = device
 
     def forward(self, input):
         hidden = self.init_hidden()
         embedded = self.embedding(input).view(1, -1, self.hidden_size)
         output = embedded
-        output, hidden = self.gru(output, hidden)
+        output, hidden = self.lstm(output, hidden)
         return output, hidden
 
     def init_hidden(self):