Fixes for Pytorch CEDR LR and hgf output - Githubissues

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

Fixes for Pytorch CEDR LR and hgf output #126

Closed andrewyates closed 3 years ago

andrewyates commented 3 years ago

Fix Pytorch CEDR with BERT weights
Fix Pytorch LR schedule after iters/steps change
Enable TF similarity matrix padding
Add smart_open dependency
Change spacy dependency to <3.0