explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-13 05:07:03 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix minibatching

Browse Source
This commit is contained in:
Matthew Honnibal 2017-07-22 20:14:49 +02:00
parent ded0df5e2f
commit 9bae0ddc50
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/gold.pyx
View File

@ -148,7 +148,7 @@ def minibatch(items, size=8):
''' '''
items = iter(items) items = iter(items)
while True: while True:
batch_size = next(size) if hasattr(size, '__next__') else size batch_size = next(size) #if hasattr(size, '__next__') else size
batch = list(cytoolz.take(int(batch_size), items)) batch = list(cytoolz.take(int(batch_size), items))
if len(batch) == 0: if len(batch) == 0:
break break