Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Set batch size in begin_training

This commit is contained in:
Matthw Honnibal 2020-07-01 15:16:59 +02:00
parent 1f7709e9a6
commit 7734cbc34d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/syntax/nn_parser.pyx
View File

@ -449,7 +449,7 @@ cdef class Parser:
if component is self: if component is self:
break break
if hasattr(component, "pipe"): if hasattr(component, "pipe"):
doc_sample = list(component.pipe(doc_sample)) doc_sample = list(component.pipe(doc_sample, batch_size=8))
else: else:
doc_sample = [component(doc) for doc in doc_sample] doc_sample = [component(doc) for doc in doc_sample]
if doc_sample: if doc_sample: