diff --git a/spacy/syntax/nn_parser.pyx b/spacy/syntax/nn_parser.pyx index 3ea17f2fe..e2dc35966 100644 --- a/spacy/syntax/nn_parser.pyx +++ b/spacy/syntax/nn_parser.pyx @@ -262,8 +262,8 @@ cdef class Parser: upper.is_noop = True else: upper = chain( - clone(Maxout(hidden_width), (depth-1)), - zero_init(Affine(nr_class, drop_factor=0.0)) + clone(Maxout(hidden_width), depth-1), + zero_init(Affine(nr_class, hidden_width, drop_factor=0.0)) ) upper.is_noop = False # TODO: This is an unfortunate hack atm!