Update spacy/ml/models/spancat.py

This commit is contained in:
Adriane Boyd 2022-05-25 13:05:41 +02:00 committed by GitHub
parent 015050f42c
commit f75a528787
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -30,7 +30,7 @@ def build_mean_max_reducer(hidden_size: int) -> Model[Ragged, Floats2d]:
reduce_max(), reduce_max(),
), ),
Maxout(nO=hidden_size, normalize=True, dropout=0.0), Maxout(nO=hidden_size, normalize=True, dropout=0.0),
) )
@registry.architectures("spacy.SpanCategorizer.v1") @registry.architectures("spacy.SpanCategorizer.v1")