From 47969e306b681f6c5630d10db36f02bd32addaf7 Mon Sep 17 00:00:00 2001 From: ines Date: Mon, 6 Nov 2017 02:40:44 +0100 Subject: [PATCH] Update example --- examples/training/train_pytorch_textcat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/training/train_pytorch_textcat.py b/examples/training/train_pytorch_textcat.py index 40a3d7c5f..1d08dc60b 100644 --- a/examples/training/train_pytorch_textcat.py +++ b/examples/training/train_pytorch_textcat.py @@ -40,7 +40,7 @@ def main(model=None, output_dir=None, n_iter=20, n_texts=2000): # Create the PyTorch neural network model, and wrap it with Thinc. This # gives it the API that spaCy expects. pt_model = create_model() - textcat = thinc.extra.wrappers.PytorchWrapper(pt_model) + textcat = thinc.extra.wrappers.PyTorchWrapper(pt_model) nlp.add_pipe(textcat, last=True) # add label to text classifier