diff --git a/beginner_source/nlp/advanced_tutorial.py b/beginner_source/nlp/advanced_tutorial.py index d175cab3749..424ca4ebc41 100644 --- a/beginner_source/nlp/advanced_tutorial.py +++ b/beginner_source/nlp/advanced_tutorial.py @@ -180,8 +180,8 @@ def __init__(self, vocab_size, tag_to_ix, embedding_dim, hidden_dim): self.hidden = self.init_hidden() def init_hidden(self): - return (autograd.Variable(torch.randn(2, 1, self.hidden_dim)), - autograd.Variable(torch.randn(2, 1, self.hidden_dim))) + return (autograd.Variable(torch.randn(2, 1, self.hidden_dim // 2)), + autograd.Variable(torch.randn(2, 1, self.hidden_dim // 2))) def _forward_alg(self, feats): # Do the forward algorithm to compute the partition function @@ -217,7 +217,7 @@ def _forward_alg(self, feats): def _get_lstm_features(self, sentence): self.hidden = self.init_hidden() embeds = self.word_embeds(sentence).view(len(sentence), 1, -1) - lstm_out, self.hidden = self.lstm(embeds) + lstm_out, self.hidden = self.lstm(embeds, self.hidden) lstm_out = lstm_out.view(len(sentence), self.hidden_dim) lstm_feats = self.hidden2tag(lstm_out) return lstm_feats