search

MIND-Lab / OCTIS

OCTIS: Comparing Topic Models is Simple! A python package to optimize and evaluate topic models (accepted at EACL2021 demo track)
MIT License
734 stars 106 forks source link

Specify GPU ID in OCTIS #95

Open ankur287 opened 1 year ago

ankur287 commented 1 year ago

How do you pass the GPU id to OCTIS? Looks like its defaulting to 0. There should be a way to specify device ID.

Thanks