Georgetown-IR-Lab / cedr

Code for CEDR: Contextualized Embeddings for Document Ranking, accepted at SIGIR 2019.
MIT License
155 stars 28 forks source link

Update train.py #29

Open tonellotto opened 4 years ago

cmacdonald commented 4 years ago

in my experience, tqdm.autonotebook isn't as auto as the name would imply. Works well on colab, but not so much on a fresh Jupyter/Jupyterlab install.