search

cemoody / lda2vec

MIT License
3.15k stars 629 forks source link

Requires cuda #26

Closed PavlosMelissinos closed 8 years ago

PavlosMelissinos commented 8 years ago

Are there any plans to run exclusively on a cpu?

cemoody commented 8 years ago

You should be able to modify the lda2vec_run.py pretty quickly. Removing the model.to_gpu() is likely the only thing you need to do.

PavlosMelissinos commented 8 years ago

Indeed, removing every reference to gpu or cuda seems to work.

Thanks for making this publicly available.