Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Pretty-print config validation errors

This commit is contained in:
Ines Montani 2020-07-10 20:01:20 +02:00
parent a3667394b4
commit defe1e7213
1 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
spacy/cli/train.py
View File

@ -6,9 +6,11 @@ from pathlib import Path
from wasabi import msg from wasabi import msg
import thinc import thinc
import thinc.schedules import thinc.schedules
from thinc.config import ConfigValidationError
from thinc.api import use_pytorch_for_gpu_memory, require_gpu, fix_random_seed from thinc.api import use_pytorch_for_gpu_memory, require_gpu, fix_random_seed
import random import random
import typer import typer
import sys
from ._util import app, Arg, Opt, parse_config_overrides from ._util import app, Arg, Opt, parse_config_overrides
from ..gold import Corpus, Example from ..gold import Corpus, Example
@ -81,12 +83,17 @@ def train(
) -> None: ) -> None:
msg.info(f"Loading config from: {config_path}") msg.info(f"Loading config from: {config_path}")
# Read the config first without creating objects, to get to the original nlp_config # Read the config first without creating objects, to get to the original nlp_config
config = util.load_config( try:
config_path, config = util.load_config(
create_objects=False, config_path,
schema=ConfigSchema, create_objects=False,
overrides=config_overrides, schema=ConfigSchema,
) overrides=config_overrides,
)
except ConfigValidationError as e:
msg.fail("Config validation error")
print(str(e).replace("Config validation error", "").strip())
sys.exit(1)
use_gpu = config["training"]["use_gpu"] use_gpu = config["training"]["use_gpu"]
if use_gpu >= 0: if use_gpu >= 0:
msg.info(f"Using GPU: {use_gpu}") msg.info(f"Using GPU: {use_gpu}")