explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 09:26:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

add pooling to NEL's TransformerListener

Browse Source
This commit is contained in:
svlandeg 2020-09-23 09:24:28 +02:00
parent 085a1c8e2b
commit 556f3e4652
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/cli/templates/quickstart_training.jinja
View File

@ -107,6 +107,9 @@ nO = null
[components.entity_linker.model.tok2vec] [components.entity_linker.model.tok2vec]
@architectures = "spacy-transformers.TransformerListener.v1" @architectures = "spacy-transformers.TransformerListener.v1"
grad_factor = 1.0 grad_factor = 1.0
[components.entity_linker.model.tok2vec.pooling]
@layers = "reduce_mean.v1"
{% endif -%} {% endif -%}
{% if "textcat" in components %} {% if "textcat" in components %}