search

capreolus-ir / capreolus

A toolkit for end-to-end neural ad hoc retrieval
https://capreolus.ai
Apache License 2.0
95 stars 32 forks source link

Fix pytorch best_weights regression #109

Closed andrewyates closed 4 years ago

andrewyates commented 4 years ago

Fixes dev weight selection in the PyTorch trainer.

Broken here: https://github.com/capreolus-ir/capreolus/commit/454e77a08efdf1d1addff90d5bae3e87f533f2f6#diff-86f8624abbb48da4b7414e84b5c0275bd469de6048ba4073d87397683238fc3c