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

Fail with an error message when trying to speed bench empty corpus

Browse Source
This commit is contained in:
Daniël de Kok 2023-01-12 09:26:25 +01:00
parent d5a99b192c
commit 2387802ee7
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/cli/benchmark_speed.py
View File

@ -6,6 +6,7 @@ from pathlib import Path
import time import time
from tqdm import tqdm from tqdm import tqdm
import typer import typer
from wasabi import msg
from .. import util from .. import util
from ..language import Language from ..language import Language
@ -41,6 +42,9 @@ def benchmark_speed_cli(
corpus = Corpus(data_path) corpus = Corpus(data_path)
docs = [eg.predicted for eg in corpus(nlp)] docs = [eg.predicted for eg in corpus(nlp)]
if len(docs) == 0:
msg.fail("Cannot benchmark speed using an empty corpus.", exits=1)
print(f"Warming up for {warmup_epochs} epochs...") print(f"Warming up for {warmup_epochs} epochs...")
warmup(nlp, docs, warmup_epochs, batch_size) warmup(nlp, docs, warmup_epochs, batch_size)