spaCy/spacy/ml/models
Daniël de Kok 2dbb332cea
TextCatParametricAttention.v1: set key transform dimensions (#13249)
* TextCatParametricAttention.v1: set key transform dimensions

This is necessary for tok2vec implementations that initialize
lazily (e.g. curated transformers).

* Add lazily-initialized tok2vec to simulate transformers

Add a lazily-initialized tok2vec to the tests and test the current
textcat models with it.

Fix some additional issues found using this test.

* isort

* Add `test.` prefix to `LazyInitTok2Vec.v1`
2024-02-02 13:01:59 +01:00
..
__init__.py
entity_linker.py
multi_task.py
parser.py
span_finder.py
spancat.py
tagger.py
textcat.py TextCatParametricAttention.v1: set key transform dimensions (#13249) 2024-02-02 13:01:59 +01:00
tok2vec.py