explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-05 12:50:20 +03:00
Code Issues Packages Projects Releases Wiki Activity

pass nO through

Browse Source
This commit is contained in:
svlandeg 2021-10-29 17:38:11 +02:00
parent 1cc0d05812
commit 87cf72d1c8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/ml/tb_framework.py
View File

@ -1,5 +1,5 @@
from typing import List, Tuple, Any, Optional from typing import List, Tuple, Any, Optional
from thinc.api import Ops, Model, normal_init, chain, list2array, Linear from thinc.api import Model, normal_init, chain, list2array, Linear
from thinc.types import Floats1d, Floats2d, Floats3d, Ints2d, Floats4d from thinc.types import Floats1d, Floats2d, Floats3d, Ints2d, Floats4d
import numpy import numpy
from ..tokens.doc import Doc from ..tokens.doc import Doc
@ -36,12 +36,12 @@ def TransitionModel(
params={ params={
"lower_W": None, # Floats2d W for the hidden layer "lower_W": None, # Floats2d W for the hidden layer
"lower_b": None, # Floats1d bias for the hidden layer "lower_b": None, # Floats1d bias for the hidden layer
"lower_pad": None, # Floats1d bias for the hidden layer "lower_pad": None, # Floats1d padding for the hidden layer
"upper_W": None, # Floats2d W for the output layer "upper_W": None, # Floats2d W for the output layer
"upper_b": None, # Floats1d bias for the output layer "upper_b": None, # Floats1d bias for the output layer
}, },
dims={ dims={
"nO": None, # Output size "nO": nO,
"nP": maxout_pieces, "nP": maxout_pieces,
"nH": hidden_width, "nH": hidden_width,
"nI": tok2vec_projected.maybe_get_dim("nO"), "nI": tok2vec_projected.maybe_get_dim("nO"),