Small hack in Language.update to make torch work

2018-09-13 22:51:52 +00:00 · 2018-09-13 22:51:52 +00:00 · a26fe8e7bb
parent 445b81ce3f
commit a26fe8e7bb
1 changed files with 4 additions and 0 deletions
--- a/spacy/language.py
+++ b/spacy/language.py
@ -426,6 +426,10 @@ class Language(object):
        def get_grads(W, dW, key=None):
            grads[key] = (W, dW)

+        get_grads.alpha = sgd.alpha
+        get_grads.b1 = sgd.b1
+        get_grads.b2 = sgd.b2
+
        pipes = list(self.pipeline)
        random.shuffle(pipes)
        for name, proc in pipes: