Skip to content

Commit e5d1b46

Browse files
authored
Merge pull request #63 from Varal7/master
Fix LSTM model in sequence_models_tutorial.py
2 parents e28cf87 + d598779 commit e5d1b46

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/nlp/sequence_models_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def init_hidden(self):
175175

176176
def forward(self, sentence):
177177
embeds = self.word_embeddings(sentence)
178-
lstm_out, self.hidden = self.lstm(embeds.view(len(sentence), 1, -1))
178+
lstm_out, self.hidden = self.lstm(embeds.view(len(sentence), 1, -1), self.hidden)
179179
tag_space = self.hidden2tag(lstm_out.view(len(sentence), -1))
180180
tag_scores = F.log_softmax(tag_space)
181181
return tag_scores

0 commit comments

Comments
 (0)