explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-02-21 22:10:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Richard Liaw 2020-07-06 16:03:26 -07:00
parent 8eb2484504
commit e061301a98
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/cli/train.py
View File

@ -172,7 +172,7 @@ def train_cli(
distributed_setup_and_train(use_gpu, num_workers, strategy, ray_address, train_args) distributed_setup_and_train(use_gpu, num_workers, strategy, ray_address, train_args)
else: else:
if use_gpu >= 0: if use_gpu >= 0:
msg.info(f"Using GPU: {use_gpu}") msg.info(f"Using GPU: {str(use_gpu)}")
util.use_gpu(use_gpu) util.use_gpu(use_gpu)
else: else:
msg.info("Using CPU") msg.info("Using CPU")
@ -602,7 +602,7 @@ def verify_cli_args(
if num_workers and num_workers > 1: if num_workers and num_workers > 1:
try: try:
from spacy_ray import distributed_setup_and_train import spacy_ray
except ImportError: except ImportError:
msg.fail("Need to `pip install spacy_ray` to use distributed training!", exits=1) msg.fail("Need to `pip install spacy_ray` to use distributed training!", exits=1)