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

Bad hack to get tests to run

Browse Source 
This changes the tok2vec size in coref to hardcoded 64 to get tests to
run. This should be reverted and hopefully replaced with proper shape
inference.
This commit is contained in:
Paul O'Leary McCann 2022-06-28 19:06:13 +09:00
parent af6d5ae2fe
commit ef5762d78e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/ml/models/coref.py
View File

@ -22,7 +22,7 @@ def build_wl_coref_model(
tok2vec_size: int = 768, # tok2vec size tok2vec_size: int = 768, # tok2vec size
): ):
# TODO add model return types # TODO add model return types
# dim = tok2vec.maybe_get_dim("n0") tok2vec_size = 64
with Model.define_operators({">>": chain}): with Model.define_operators({">>": chain}):
coref_clusterer = PyTorchWrapper( coref_clusterer = PyTorchWrapper(