Fix new feature flags

This commit is contained in:
Matthew Honnibal 2018-08-27 02:12:39 +02:00
parent 598dbf1ce0
commit 4d2d7d5866
2 changed files with 2 additions and 2 deletions

View File

@ -263,7 +263,7 @@ def Tok2Vec(width, embed_size, **kwargs):
shape = HashEmbed(width, embed_size//2, column=cols.index(SHAPE),
name='embed_shape')
else:
prefix, suffix, shape = None
prefix, suffix, shape = (None, None, None)
if pretrained_vectors is not None:
glove = StaticVectors(pretrained_vectors, width, column=cols.index(ID))

View File

@ -292,7 +292,7 @@ def initialize_pipeline(nlp, docs, golds, config, device):
nlp.tagger.add_label(tag)
return nlp.begin_training(
lambda: golds_to_gold_tuples(docs, golds), device=device,
subword_features=config.subword_features, config.conv_depth=conv_depth)
subword_features=config.subword_features, conv_depth=config.conv_depth)
########################