From 7734cbc34dd9510baeb28d256cd921d80ad76e12 Mon Sep 17 00:00:00 2001 From: Matthw Honnibal Date: Wed, 1 Jul 2020 15:16:59 +0200 Subject: [PATCH] Set batch size in begin_training --- spacy/syntax/nn_parser.pyx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/syntax/nn_parser.pyx b/spacy/syntax/nn_parser.pyx index ceaea3c9c..743b4ca1d 100644 --- 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: