Add --no-gpu flag

facebookresearch · Aug 12, 2018 · 670168a · 670168a
1 parent e59bc5c
commit 670168a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/examples/eval_saved.py b/examples/eval_saved.py
@@ -37,6 +37,8 @@ def main():
                         help='a NumPy binary file containing the corresponding embeddings')
     parser.add_argument('-t', '--tasks',
                         help='a comma-separated list of tasks')
+    parser.add_argument('--no-gpu', action='store_true',
+                        help='do not use GPU (turn off PyTorch)')
     args = parser.parse_args()
 
     sent2emb = {}
@@ -70,7 +72,7 @@ def batcher(params, batch):
         return embeddings
 
     params_senteval = {
-        'task_path': PATH_TO_DATA, 'usepytorch': True, 'kfold': 10
+        'task_path': PATH_TO_DATA, 'usepytorch': not args.no_gpu, 'kfold': 10
     }
     params_senteval['classifier'] = {
         'nhid': 0, 'optim': 'adam', 'batch_size': 64, 'tenacity': 5,