Set batch size in begin_training

2020-07-01 15:16:59 +02:00 · 2020-07-01 15:16:59 +02:00 · 7734cbc34d
parent 1f7709e9a6
commit 7734cbc34d
1 changed files with 1 additions and 1 deletions
--- a/spacy/syntax/nn_parser.pyx
+++ b/spacy/syntax/nn_parser.pyx
@ -449,7 +449,7 @@ cdef class Parser:
                if component is self:
                    break
                if hasattr(component, "pipe"):
-                    doc_sample = list(component.pipe(doc_sample))
+                    doc_sample = list(component.pipe(doc_sample, batch_size=8))
                else:
                    doc_sample = [component(doc) for doc in doc_sample]
        if doc_sample: