castorini / bertserini

BERTserini
https://github.com/castorini/bertserini
Apache License 2.0
25 stars 10 forks source link

Code freeze on inference #14

Closed code158m closed 2 years ago

code158m commented 3 years ago

Hi, When I run the inference code , it freezes almost each time. No error comes, but the code freezes randomly. Is this a common issue? Any suggestions on how to fix this? Thanks!

amyxie361 commented 2 years ago

Might related to the transformers version issue. Try roll back it to transformers==3.4.0