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(),
),
Maxout(nO=hidden_size, normalize=True, dropout=0.0),
)
)
@registry.architectures("spacy.SpanCategorizer.v1")