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

Disable formatting of benchmark speed arguments

Browse Source
This commit is contained in:
Daniël de Kok 2023-01-12 09:05:39 +01:00
parent c2f8c65648
commit d5a99b192c
1 changed files with 6 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
spacy/cli/benchmark_speed.py
View File

@ -19,25 +19,16 @@ from ._util import Arg, Opt, benchmark_cli, setup_gpu
context_settings={"allow_extra_args": True, "ignore_unknown_options": True}, context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
) )
def benchmark_speed_cli( def benchmark_speed_cli(
# fmt: off
ctx: typer.Context, ctx: typer.Context,
model: str = Arg(..., help="Model name or path"), model: str = Arg(..., help="Model name or path"),
data_path: Path = Arg( data_path: Path = Arg(..., help="Location of binary evaluation data in .spacy format", exists=True),
..., help="Location of binary evaluation data in .spacy format", exists=True batch_size: Optional[int] = Opt(None, "--batch-size", "-b", min=1, help="Override the pipeline batch size"),
),
batch_size: Optional[int] = Opt(
None, "--batch-size", "-b", min=1, help="Override the pipeline batch size"
),
no_shuffle: bool = Opt(False, "--no-shuffle", help="Do not shuffle benchmark data"), no_shuffle: bool = Opt(False, "--no-shuffle", help="Do not shuffle benchmark data"),
use_gpu: int = Opt(-1, "--gpu-id", "-g", help="GPU ID or -1 for CPU"), use_gpu: int = Opt(-1, "--gpu-id", "-g", help="GPU ID or -1 for CPU"),
n_batches: int = Opt( n_batches: int = Opt(50, "--batches", help="Minimum number of batches to benchmark", min=30,),
50, warmup_epochs: int = Opt(3, "--warmup", "-w", min=0, help="Number of iterations over the data for warmup"),
"--batches", # fmt: on
help="Minimum number of batches to benchmark",
min=30,
),
warmup_epochs: int = Opt(
3, "--warmup", "-w", min=0, help="Number of iterations over the data for warmup"
),
): ):
""" """
Benchmark a pipeline. Expects a loadable spaCy pipeline and benchmark Benchmark a pipeline. Expects a loadable spaCy pipeline and benchmark