What is the training benchmark for model `BERTopic_Wikipedia`

#3
by benjaminliupenrose - opened

Hello, I am trying to rerun the training for model with more customized training data. May I know for your training, what is

total training time
GPU spec and numbers
CPU spec and numbers
RAM
Thank you for giving us advice!

Thank you for your question! It was a while ago but I believe it took a couple of hours using a T4 GPU on a free google colab session. You can find a number of guides also here: https://github.com/MaartenGr/BERTopic?tab=readme-ov-file#getting-started

I believe I used the "Topic Modeling on Large Data" notebook.

Sign up or log in to comment