search

texttron / tevatron

Tevatron - A flexible toolkit for neural retrieval research and development.
http://tevatron.ai
Apache License 2.0
494 stars 94 forks source link

Fix average pooling in dense model #121

Closed silencio94 closed 4 months ago

silencio94 commented 4 months ago

fixed incorrect average pooling

refer: https://github.com/facebookresearch/contriever/blob/39fb2201450cdba1648183737ed56d6b1bc33778/src/contriever.py#L46