explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 18:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix GPU training for textcat. Closes #3473

Browse Source
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

4
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 = (
with_cpu(Model.ops,
extract_ngrams(ngram_size, attr=ORTH) extract_ngrams(ngram_size, attr=ORTH)
>> with_cpu(Model.ops, >> LinearModel(nr_class)
LinearModel(nr_class)
) )
) )
if not no_output_layer: if not no_output_layer: