From 27ee0fff3ce5f9d1c1169061d49200745ea3ba4e Mon Sep 17 00:00:00 2001
From: thomwolf <thomwolf@gmail.com>
Date: Sat, 17 Nov 2018 23:04:44 +0100
Subject: [PATCH] add no_cuda args in extract_features

---
 examples/extract_features.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/examples/extract_features.py b/examples/extract_features.py
index 84e9a8d6ab..b24d6c9ad2 100644
--- a/examples/extract_features.py
+++ b/examples/extract_features.py
@@ -208,6 +208,10 @@ def main():
                         type=int,
                         default=-1,
                         help = "local_rank for distributed training on gpus")
+    parser.add_argument("--no_cuda",
+                        default=False,
+                        action='store_true',
+                        help="Whether not to use CUDA when available")
 
     args = parser.parse_args()