add no_cuda args in extract_features

This commit is contained in:
thomwolf 2018-11-17 23:04:44 +01:00
parent aa50fd196f
commit 27ee0fff3c
1 changed files with 4 additions and 0 deletions

View File

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