From 892f2552e0e7ff81ccf0cbc19e66e546e4f8d07a Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Thu, 2 Jul 2020 20:07:52 +0200 Subject: [PATCH] Fix pretrain args --- spacy/cli/pretrain.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/spacy/cli/pretrain.py b/spacy/cli/pretrain.py index 6eda4ca7e..c2d086fb6 100644 --- a/spacy/cli/pretrain.py +++ b/spacy/cli/pretrain.py @@ -87,12 +87,11 @@ def pretrain( output_dir, width=96, conv_depth=4, - bilstm_depth=0, cnn_pieces=3, sa_depth=0, - use_chars=False, cnn_window=1, bilstm_depth=0, + use_chars=False, embed_rows=2000, loss_func="cosine", use_vectors=False,