From fe0cdcd461f4c77db4ce120b3dee2d960d83d605 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 28 Jul 2020 13:59:46 +0200 Subject: [PATCH] Fixes --- spacy/ml/models/tok2vec.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/spacy/ml/models/tok2vec.py b/spacy/ml/models/tok2vec.py index 4bcd61625..4c4bd0d22 100644 --- a/spacy/ml/models/tok2vec.py +++ b/spacy/ml/models/tok2vec.py @@ -50,7 +50,8 @@ def HashEmbed( width, rows if feature == NORM else rows // 2, column=cols.index(feature), - seed=seed + seed=seed, + dropout=0.0 ) if also_embed_subwords: @@ -67,14 +68,14 @@ def HashEmbed( model = chain( concatenate( chain(FeatureExtractor(cols), concatenate(*embeddings)), - StaticVectors(width, dropout=dropout) + StaticVectors(width, dropout=0.0) ), - Maxout(width, dropout=dropout, normalize=True) + Maxout(width, pieces=3, dropout=0.0, normalize=True) ) else: model = chain( chain(FeatureExtractor(cols), concatenate(*embeddings)), - Maxout(width, concat_size, dropout=dropout, normalize=True) + Maxout(width, pieces=3, dropout=0.0, normalize=True) ) return model