Fix GPU training for textcat. Closes #3473

2019-03-26 13:36:11 +01:00 · 2019-03-26 13:36:11 +01:00 · f77bf2bdb1
parent 85dcd9477e
commit f77bf2bdb1
1 changed files with 3 additions and 3 deletions
--- a/spacy/_ml.py
+++ b/spacy/_ml.py
@ -614,9 +614,9 @@ def build_bow_text_classifier(nr_class, ngram_size=1, exclusive_classes=False,
        no_output_layer=False, **cfg):
    with Model.define_operators({">>": chain}):
        model = (
-            extract_ngrams(ngram_size, attr=ORTH) 
-            >> with_cpu(Model.ops,
-                LinearModel(nr_class)
+            with_cpu(Model.ops,
+                extract_ngrams(ngram_size, attr=ORTH) 
+                >> LinearModel(nr_class)
            )
        )
        if not no_output_layer: