Fix issue #3112 part 1

This commit is contained in:
Matthew Honnibal 2019-02-21 09:27:38 +01:00
parent b21481eeca
commit 6574e4f2d3

View File

@ -261,8 +261,9 @@ class ParserStepModel(Model):
# lowest score.
# numpy's nan_to_num function doesn't take a value, and nan is replaced
# by 0...-inf is replaced by minimum, so we go via that. Ugly to the max.
scores[self.ops.xp.isnan(scores)] = -self.ops.xp.inf
self.ops.xp.nan_to_num(scores, copy=False)
# Note that scores is always a numpy array! Should fix #3112
scores[numpy.isnan(scores)] = -numpy.inf
numpy.nan_to_num(scores, copy=False)
def backprop_parser_step(d_scores, sgd=None):
# If we have a non-zero gradient for a previously unseen class,