Remove for double usage of log_softmax (pytorch#721)

KernLC · Mar 5, 2020 · 1800513 · 1800513
1 parent e9e7672
commit 1800513
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/word_language_model/main.py b/word_language_model/main.py
@@ -101,7 +101,7 @@ def batchify(data, bsz):
 else:
     model = model.RNNModel(args.model, ntokens, args.emsize, args.nhid, args.nlayers, args.dropout, args.tied).to(device)
 
-criterion = nn.CrossEntropyLoss()
+criterion = nn.NLLLoss()
 
 ###############################################################################
 # Training code