Skip to content

Commit ccb38b4

Browse files
author
Svetlana Karslioglu
authored
Update text_sentiment_ngrams_tutorial.py (#2246)
Fixes #2245
1 parent f325d12 commit ccb38b4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/text_sentiment_ngrams_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ class TextClassificationModel(nn.Module):
148148

149149
def __init__(self, vocab_size, embed_dim, num_class):
150150
super(TextClassificationModel, self).__init__()
151-
self.embedding = nn.EmbeddingBag(vocab_size, embed_dim, sparse=True)
151+
self.embedding = nn.EmbeddingBag(vocab_size, embed_dim, sparse=False)
152152
self.fc = nn.Linear(embed_dim, num_class)
153153
self.init_weights()
154154

0 commit comments

Comments
 (0)