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

Proactively remove unused listeners

Browse Source 
With this the changes in initialize.py might be unecessary.

Requires testing.
This commit is contained in:
Paul O'Leary McCann 2021-03-17 22:41:41 +09:00
parent ef77c88638
commit 40bc01e668
2 changed files with 24 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spacy/language.py
View File

@ -1686,8 +1686,14 @@ class Language:
) )
# Detect components with listeners that are not frozen consistently # Detect components with listeners that are not frozen consistently
for name, proc in nlp.pipeline: for name, proc in nlp.pipeline:
if getattr(proc, "listening_components", None): # e.g. tok2vec/transformer # Remove listeners not in the pipeline
for listener in proc.listening_components: listener_names = getattr(proc, "listening_components", [])
unused_listener_names = [ll for ll in listener_names if ll not in nlp.pipe_names]
for listener_name in unused_listener_names:
for listener in proc.listener_map.get(listener_name, []):
proc.remove_listener(listener, listener_name)
for listener in getattr(proc, "listening_components", []): # e.g. tok2vec/transformer
# If it's a component sourced from another pipeline, we check if # If it's a component sourced from another pipeline, we check if
# the tok2vec listeners should be replaced with standalone tok2vec # the tok2vec listeners should be replaced with standalone tok2vec
# models (e.g. so component can be frozen without its performance # models (e.g. so component can be frozen without its performance

5
spacy/training/initialize.py
View File

@ -72,10 +72,9 @@ def init_nlp(config: Config, *, use_gpu: int = -1) -> "Language":
logger.info(f"Initialized pipeline components: {nlp.pipe_names}") logger.info(f"Initialized pipeline components: {nlp.pipe_names}")
# Detect components with listeners that are not frozen consistently # Detect components with listeners that are not frozen consistently
for name, proc in nlp.pipeline: for name, proc in nlp.pipeline:
if getattr(proc, "listening_components", None): # e.g. tok2vec/transformer for listener in getattr(proc, "listening_components", []): # e.g. tok2vec/transformer
for listener in proc.listening_components:
# Don't warn about components not in the pipeline # Don't warn about components not in the pipeline
if listener not in nlp.pipeline: if listener not in nlp.pipe_names:
continue continue
if listener in frozen_components and name not in frozen_components: if listener in frozen_components and name not in frozen_components: