From 8f913a74ca4eaa2e36354530bfe8e822b1697777 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 17 Sep 2017 05:46:36 -0500 Subject: [PATCH] Fix defaults and args to build_tagger_model --- spacy/_ml.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/spacy/_ml.py b/spacy/_ml.py index 14341c407..2f6a36942 100644 --- a/spacy/_ml.py +++ b/spacy/_ml.py @@ -467,9 +467,8 @@ def getitem(i): return X[i], None return layerize(getitem_fwd) -def build_tagger_model(nr_class, token_vector_width, **cfg): - embed_size = util.env_opt('embed_size', 7500) - pretrained_dims = cfg.get('pretrained_dims', 0) +def build_tagger_model(nr_class, token_vector_width, pretrained_dims=0, **cfg): + embed_size = util.env_opt('embed_size', 4000) with Model.define_operators({'>>': chain, '+': add}): # Input: (doc, tensor) tuples private_tok2vec = Tok2Vec(token_vector_width, embed_size,