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

Fix GPU training for textcat. Closes #3473

This commit is contained in:
Matthew Honnibal 2019-03-26 13:36:11 +01:00
parent 85dcd9477e
commit f77bf2bdb1
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/_ml.py
View File

@ -614,9 +614,9 @@ def build_bow_text_classifier(nr_class, ngram_size=1, exclusive_classes=False,
no_output_layer=False, **cfg): no_output_layer=False, **cfg):
with Model.define_operators({">>": chain}): with Model.define_operators({">>": chain}):
model = ( model = (
extract_ngrams(ngram_size, attr=ORTH) with_cpu(Model.ops,
>> with_cpu(Model.ops, extract_ngrams(ngram_size, attr=ORTH)
LinearModel(nr_class) >> LinearModel(nr_class)
) )
) )
if not no_output_layer: if not no_output_layer: