Fix GPU training for textcat. Closes
@@ -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 = (
+ with_cpu(Model.ops,
extract_ngrams(ngram_size, attr=ORTH)
- >> with_cpu(Model.ops,
- LinearModel(nr_class)
+ >> LinearModel(nr_class)
)
)
if not no_output_layer:
