From f67c1735c554be4afafeffbeb10613309d4cb4d0 Mon Sep 17 00:00:00 2001 From: Paul O'Leary McCann Date: Wed, 6 Jul 2022 18:58:57 +0900 Subject: [PATCH] Remove tok2vec_size from coref --- spacy/ml/models/coref.py | 1 - spacy/pipeline/coref.py | 1 - 2 files changed, 2 deletions(-) diff --git a/spacy/ml/models/coref.py b/spacy/ml/models/coref.py index 377f02236..2b38beef7 100644 --- a/spacy/ml/models/coref.py +++ b/spacy/ml/models/coref.py @@ -22,7 +22,6 @@ def build_wl_coref_model( # pairs to keep per mention after rough scoring antecedent_limit: int = 50, antecedent_batch_size: int = 512, - tok2vec_size: int = 768, # tok2vec size ): # TODO add model return types diff --git a/spacy/pipeline/coref.py b/spacy/pipeline/coref.py index ef74a83a4..6685b112e 100644 --- a/spacy/pipeline/coref.py +++ b/spacy/pipeline/coref.py @@ -31,7 +31,6 @@ from ..scorer import Scorer default_config = """ [model] @architectures = "spacy.Coref.v1" -tok2vec_size = 768 distance_embedding_size = 20 hidden_size = 1024 depth = 1