From 42295b66ee6b81177e18d37574f2d6725aa8edff Mon Sep 17 00:00:00 2001 From: Svetlana Karslioglu Date: Fri, 10 Mar 2023 12:55:17 -0800 Subject: [PATCH] Update text_sentiment_ngrams_tutorial.py Fixes #2245 --- beginner_source/text_sentiment_ngrams_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/text_sentiment_ngrams_tutorial.py b/beginner_source/text_sentiment_ngrams_tutorial.py index 204108a5028..1dd7466344c 100644 --- a/beginner_source/text_sentiment_ngrams_tutorial.py +++ b/beginner_source/text_sentiment_ngrams_tutorial.py @@ -148,7 +148,7 @@ class TextClassificationModel(nn.Module): def __init__(self, vocab_size, embed_dim, num_class): super(TextClassificationModel, self).__init__() - self.embedding = nn.EmbeddingBag(vocab_size, embed_dim, sparse=True) + self.embedding = nn.EmbeddingBag(vocab_size, embed_dim, sparse=False) self.fc = nn.Linear(embed_dim, num_class) self.init_weights()