From 59271e887a81f02ac7b608bb71459dff5d49cfa9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ren=C3=A9=20Octavio=20Queiroz=20Dias?= Date: Sat, 6 Feb 2021 09:44:51 -0300 Subject: [PATCH] fix: TransformerListener with TextCatEnsemble (#6951) * bug: Regression test Issue #6946 * fix: Fix issue #6946 * chore: Remove regression test --- spacy/ml/models/textcat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/spacy/ml/models/textcat.py b/spacy/ml/models/textcat.py index e0c11ed99..0234530e6 100644 --- a/spacy/ml/models/textcat.py +++ b/spacy/ml/models/textcat.py @@ -107,6 +107,7 @@ def init_ensemble_textcat(model, X, Y) -> Model: model.get_ref("maxout_layer").set_dim("nO", tok2vec_width) model.get_ref("maxout_layer").set_dim("nI", tok2vec_width) model.get_ref("norm_layer").set_dim("nI", tok2vec_width) + model.get_ref("norm_layer").set_dim("nO", tok2vec_width) init_chain(model, X, Y) return model