Using GPU while processing concepts

MaartenGr / Concept

Concept Modeling: Topic Modeling on Images and Text

https://maartengr.github.io/Concept/

MIT License

192 stars 16 forks source link

Using GPU while processing concepts #9

Closed vvkishere closed 2 years ago

vvkishere commented 2 years ago

Any plans on supporting GPU processing for this to speed up the model creation?

MaartenGr commented 2 years ago

Currently, GPU is only supported for generating multi-modal embeddings. This is typically the most computation-heavy task in Concept. As of right now, there are no plans on supporting GPU processing for HDBSCAN and UMAP.

vvkishere commented 2 years ago

Thank you.