Allow training with no cuda gpus

stanford-futuredata / ColBERT

ColBERT: state-of-the-art neural search (SIGIR'20, TACL'21, NeurIPS'21, NAACL'22, CIKM'22, ACL'23, EMNLP'23)

MIT License

2.67k stars 355 forks source link

Allow training with no cuda gpus #313

Open JoshuaPurtell opened 4 months ago

JoshuaPurtell commented 4 months ago

Presently, when I attempt to train on my MacBook this code fails.

With more consumer attention via Ragatouille, I believe it'd be good to enable people to experiment with local cpu-base training - even if it's slower.