update translate.py gpu option

MingChaoSun · Feb 23, 2017 · b840cad · b840cad
1 parent a742fe2
commit b840cad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/OpenNMT/translate.py b/OpenNMT/translate.py
@@ -49,7 +49,7 @@ def reportScore(name, scoreTotal, wordsTotal):
 
 def main():
     opt = parser.parse_args()
-    opt.cuda = True if opt.gpu > -1
+    opt.cuda = opt.gpu > -1
     torch.cuda.set_device(opt.gpu)
 
     translator = onmt.Translator(opt)