From c9036a6d795bc91c543d0c004a39fab65be8aef8 Mon Sep 17 00:00:00 2001 From: Lj Miranda Date: Fri, 18 Nov 2022 13:16:33 +0800 Subject: [PATCH] Include zero_init.v1 for spancat --- spacy/pipeline/spancat_exclusive.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/spacy/pipeline/spancat_exclusive.py b/spacy/pipeline/spancat_exclusive.py index adf3741c1..75c4446af 100644 --- a/spacy/pipeline/spancat_exclusive.py +++ b/spacy/pipeline/spancat_exclusive.py @@ -18,6 +18,12 @@ spancat_exclusive_default_config = """ @architectures = "spacy.SpanCategorizer.v1" scorer = {"@layers": "Softmax.v2"} +[model.scorer.init_W] +@initializers = "zero_init.v1" + +[model.scorer.init_b] +@initializers = "zero_init.v1" + [model.reducer] @layers = spacy.mean_max_reducer.v1 hidden_size = 128