fix Runtime error in Library tutorial (OpenNMT#1399)

AMusketeer · Apr 17, 2019 · dcaeb23 · dcaeb23
1 parent db4d2fb
commit dcaeb23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/Library.md b/docs/source/Library.md
@@ -60,7 +60,7 @@ model.to(device)
 # Specify the tgt word generator and loss computation module
 model.generator = nn.Sequential(
     nn.Linear(rnn_size, len(tgt_vocab)),
-    nn.LogSoftmax(dim=-1))
+    nn.LogSoftmax(dim=-1)).to(device)
 
 loss = onmt.utils.loss.NMTLossCompute(
     criterion=nn.NLLLoss(ignore_index=tgt_padding, reduction="sum"),