From 769ecbda123c6f5986c26223747144d1e0b2e91c Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Wed, 3 Nov 2021 15:11:07 +0100 Subject: [PATCH] Fix tok2vec-less textcat generation in website quickstart (#9610) --- spacy/cli/templates/quickstart_training.jinja | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/spacy/cli/templates/quickstart_training.jinja b/spacy/cli/templates/quickstart_training.jinja index 50dbc6e42..b78806fec 100644 --- a/spacy/cli/templates/quickstart_training.jinja +++ b/spacy/cli/templates/quickstart_training.jinja @@ -16,8 +16,10 @@ gpu_allocator = null [nlp] lang = "{{ lang }}" -{%- set no_tok2vec = components|length == 1 and (("textcat" in components or "textcat_multilabel" in components) and optimize == "efficiency")-%} -{%- if not no_tok2vec and ("tagger" in components or "morphologizer" in components or "parser" in components or "ner" in components or "entity_linker" in components or "textcat" in components or "textcat_multilabel" in components) -%} +{%- set has_textcat = ("textcat" in components or "textcat_multilabel" in components) -%} +{%- set with_accuracy = optimize == "accuracy" -%} +{%- set has_accurate_textcat = has_textcat and with_accuracy -%} +{%- if ("tagger" in components or "morphologizer" in components or "parser" in components or "ner" in components or "entity_linker" in components or has_accurate_textcat) -%} {%- set full_pipeline = ["transformer" if use_transformer else "tok2vec"] + components %} {%- else -%} {%- set full_pipeline = components %} @@ -199,7 +201,7 @@ no_output_layer = false {# NON-TRANSFORMER PIPELINE #} {% else -%} -{% if not no_tok2vec-%} +{% if "tok2vec" in full_pipeline -%} [components.tok2vec] factory = "tok2vec"