fix bug in utils.py (#163)

Author: ThomasDelteil

scripts/word_embeddings/utils.py

@@ -40,6 +40,8 @@ def numba_njit(func):
 def get_context(args):
     if args.gpu is None or args.gpu == '':
         context = [mx.cpu()]
+    elif isinstance(args.gpu, int):
+        context = [mx.gpu(args.gpu)]
     else:
         context = [mx.gpu(int(i)) for i in args.gpu]
     return context