Set softmax attr in tagger model

This commit is contained in:
Matthew Honnibal 2017-11-03 11:22:01 +01:00
parent a5b05f85f0
commit c9b118a7e9

View File

@ -416,7 +416,7 @@ def build_tagger_model(nr_class, **cfg):
)
model.nI = None
model.tok2vec = tok2vec
model.softmax
model.softmax = softmax
return model