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

Check SGD is not None in update

Browse Source
This commit is contained in:
Matthew Honnibal 2017-08-14 12:09:18 +02:00
parent 6a42cc16ff
commit ac6c25f762
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/_ml.py
View File

@ -381,7 +381,8 @@ def fine_tune(embedding, combine=None):
flat_grad = model.ops.flatten(d_output) flat_grad = model.ops.flatten(d_output)
model.d_mix[1] += flat_tokvecs.dot(flat_grad.T).sum() model.d_mix[1] += flat_tokvecs.dot(flat_grad.T).sum()
model.d_mix[0] += flat_vecs.dot(flat_grad.T).sum() model.d_mix[0] += flat_vecs.dot(flat_grad.T).sum()
sgd(model._mem.weights, model._mem.gradient, key=model.id) if sgd is not None:
sgd(model._mem.weights, model._mem.gradient, key=model.id)
return d_output return d_output
return output, fine_tune_bwd return output, fine_tune_bwd
model = wrap(fine_tune_fwd, embedding) model = wrap(fine_tune_fwd, embedding)