Switch the model to evaluation mode before generation

ownway22 · Mar 19, 2017 · b8cacb0 · b8cacb0
1 parent 42e5b99
commit b8cacb0
Showing 1 changed file with 1 addition and 0 deletions.
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()