mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-18 12:12:20 +03:00
Remove tok2vec_size from coref
This commit is contained in:
parent
bd17c38b74
commit
f67c1735c5
|
@ -22,7 +22,6 @@ def build_wl_coref_model(
|
||||||
# pairs to keep per mention after rough scoring
|
# pairs to keep per mention after rough scoring
|
||||||
antecedent_limit: int = 50,
|
antecedent_limit: int = 50,
|
||||||
antecedent_batch_size: int = 512,
|
antecedent_batch_size: int = 512,
|
||||||
tok2vec_size: int = 768, # tok2vec size
|
|
||||||
):
|
):
|
||||||
# TODO add model return types
|
# TODO add model return types
|
||||||
|
|
||||||
|
|
|
@ -31,7 +31,6 @@ from ..scorer import Scorer
|
||||||
default_config = """
|
default_config = """
|
||||||
[model]
|
[model]
|
||||||
@architectures = "spacy.Coref.v1"
|
@architectures = "spacy.Coref.v1"
|
||||||
tok2vec_size = 768
|
|
||||||
distance_embedding_size = 20
|
distance_embedding_size = 20
|
||||||
hidden_size = 1024
|
hidden_size = 1024
|
||||||
depth = 1
|
depth = 1
|
||||||
|
|
Loading…
Reference in New Issue
Block a user