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()