Switch the model to evaluation mode before generation

Henderake · soumith · commit b8cacb0e75cc · 2017-03-18T19:32:41.000-05:00
diff --git a/word_language_model/generate.py b/word_language_model/generate.py
@@ -46,6 +46,7 @@
 
 with open(args.checkpoint, 'rb') as f:
     model = torch.load(f)
+model.eval()
 
 if args.cuda:
     model.cuda()
