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

Remove 'rebatch' op, and remove min-batch cap

This commit is contained in:
Matthew Honnibal 2017-05-19 18:16:36 -05:00
parent 3376d4d6e8
commit e84de028b5
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/syntax/nn_parser.pyx
View File

@ -229,8 +229,6 @@ cdef class Parser:
nI=token_vector_width, nI=token_vector_width,
pieces=maxout_pieces) pieces=maxout_pieces)
lower = rebatch(1024, lower)
with Model.use_device('cpu'): with Model.use_device('cpu'):
upper = chain( upper = chain(
Maxout(hidden_width), Maxout(hidden_width),
@ -342,8 +340,7 @@ cdef class Parser:
backprops = [] backprops = []
cdef float loss = 0. cdef float loss = 0.
cutoff = max(1, len(todo) // 10) while todo:
while len(todo) >= cutoff:
states, golds = zip(*todo) states, golds = zip(*todo)
token_ids = self.get_token_ids(states) token_ids = self.get_token_ids(states)