Fix default token_vector_width

This commit is contained in:
Matthew Honnibal 2018-11-30 16:40:11 +00:00
parent 681258e29b
commit 1b240f2119
1 changed files with 1 additions and 1 deletions

View File

@ -442,7 +442,7 @@ class Tensorizer(Pipe):
**cfg: Config parameters. **cfg: Config parameters.
RETURNS (Model): A `thinc.neural.Model` or similar instance. RETURNS (Model): A `thinc.neural.Model` or similar instance.
""" """
input_size = util.env_opt('token_vector_width', cfg.get('input_size', 128)) input_size = util.env_opt('token_vector_width', cfg.get('input_size', 96))
return zero_init(Affine(output_size, input_size, drop_factor=0.0)) return zero_init(Affine(output_size, input_size, drop_factor=0.0))
def __init__(self, vocab, model=True, **cfg): def __init__(self, vocab, model=True, **cfg):