From 790b37390136bab87126c3eb69fb9354aef5c7c5 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Tue, 23 Jun 2020 14:05:00 +0200 Subject: [PATCH] fix var --- spacy/ml/tb_framework.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/spacy/ml/tb_framework.py b/spacy/ml/tb_framework.py index 21779ddaa..88f27f0bf 100644 --- a/spacy/ml/tb_framework.py +++ b/spacy/ml/tb_framework.py @@ -39,7 +39,8 @@ def forward(model, X, is_train): def init(model, X=None, Y=None): model.get_ref("tok2vec").initialize(X=X) - model.get_ref("lower").initialize() + lower = model.get_ref("lower") + lower.initialize() if model.attrs["has_upper"]: statevecs = model.ops.alloc2f(2, lower.get_dim("nO")) model.get_ref("upper").initialize(X=statevecs)