Fix training with vectors

This commit is contained in:
Matthew Honnibal 2019-10-28 18:06:38 +01:00
parent a927b3a21e
commit b1505380ff
1 changed files with 1 additions and 1 deletions

View File

@ -360,7 +360,7 @@ def Tok2Vec(width, embed_size, **kwargs):
"config": { "config": {
"vectors_name": pretrained_vectors, "vectors_name": pretrained_vectors,
"width": width, "width": width,
"column": cols.index(ID), "column": cols.index("ID")
}, },
} }
if cnn_maxout_pieces >= 2: if cnn_maxout_pieces >= 2: