Fix dropout bug in beam parser

This commit is contained in:
Matthew Honnibal 2018-03-10 23:16:40 +01:00
parent 9cc202d670
commit 4b72c38556

View File

@ -525,7 +525,7 @@ cdef class Parser:
if n_states == 0: if n_states == 0:
break break
vectors, _ = state2vec.begin_update(token_ids[:n_states], drop) vectors, _ = state2vec.begin_update(token_ids[:n_states], drop)
scores, _ = vec2scores(vectors, drop) scores, _ = vec2scores.begin_update(vectors, drop=drop)
c_scores = <float*>scores.data c_scores = <float*>scores.data
for beam in todo: for beam in todo:
for i in range(beam.size): for i in range(beam.size):