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

error

Browse Source
This commit is contained in:
Richard Liaw 2020-06-22 18:54:58 -07:00
parent 2c73623a6b
commit 9987b9a5d4
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/cli/train_from_config.py
View File

@ -126,7 +126,7 @@ class ConfigSchema(BaseModel):
verbose=("Display more information for debugging purposes", "flag", "VV", bool), verbose=("Display more information for debugging purposes", "flag", "VV", bool),
use_gpu=("Use GPU", "option", "g", int), use_gpu=("Use GPU", "option", "g", int),
num_workers=("Parallel Workers", "option", "j", int), num_workers=("Parallel Workers", "option", "j", int),
strategy=("Distributed training strategy", "option", "strat", str), strategy=("Distributed training strategy (requires spacy_ray)", "option", "strat", str),
tag_map_path=("Location of JSON-formatted tag map", "option", "tm", Path), tag_map_path=("Location of JSON-formatted tag map", "option", "tm", Path),
omit_extra_lookups=("Don't include extra lookups in model", "flag", "OEL", bool), omit_extra_lookups=("Don't include extra lookups in model", "flag", "OEL", bool),
# fmt: on # fmt: on
@ -196,9 +196,9 @@ def train_cli(
if num_workers and num_workers > 1: if num_workers and num_workers > 1:
try: try:
from ray_spacy import distributed_setup_and_train from spacy_ray import distributed_setup_and_train
except ImportError: except ImportError:
msg.fail("Need to install ray_spacy to use distributed training!", exits=1) msg.fail("Need to `pip install spacy_ray` to use distributed training!", exits=1)
distributed_setup_and_train(use_gpu, num_workers, strategy, train_args) distributed_setup_and_train(use_gpu, num_workers, strategy, train_args)
else: else:
if use_gpu >= 0: if use_gpu >= 0: