fix cbow bug

L1aoXingyu · L1aoXingyu · commit 078911abaf5b · 2017-08-31T21:22:26.000+08:00
diff --git a/06-Natural Language Process/bag-of-word.py b/06-Natural Language Process/bag-of-word.py
@@ -50,7 +50,7 @@ def forward(self, x):
 if torch.cuda.is_available():
     model = model.cuda()
 
-criterion = nn.CrossEntropyLoss()
+criterion = nn.NLLLoss()
 optimizer = optim.SGD(model.parameters(), lr=1e-3)
 
 for epoch in range(100):
