diff options
author | Yuhao Zhang <zyh@stanford.edu> | 2019-01-25 12:25:55 +0300 |
---|---|---|
committer | Yuhao Zhang <zyh@stanford.edu> | 2019-01-25 12:25:55 +0300 |
commit | b012464723c63ea89ce7bfecec8a67c0adbd3817 (patch) | |
tree | 245aeda50999677b9adde88d0276ce2ed02e568e /demo | |
parent | 195959d07d1f89b1804604d1cff30b0217f5602b (diff) |
Undo changes to use_gpu argument
Diffstat (limited to 'demo')
-rw-r--r-- | demo/pipeline_demo.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/demo/pipeline_demo.py b/demo/pipeline_demo.py index be4f5687..eb08f54c 100644 --- a/demo/pipeline_demo.py +++ b/demo/pipeline_demo.py @@ -33,7 +33,7 @@ if __name__ == '__main__': # set up a pipeline print('---') print('Building pipeline...') - pipeline = stanfordnlp.Pipeline(models_dir=args.models_dir, lang=args.lang, cpu=args.cpu) + pipeline = stanfordnlp.Pipeline(models_dir=args.models_dir, lang=args.lang, use_gpu=(not args.cpu)) # process the document doc = pipeline(example_sentences[args.lang]) # access nlp annotations |