search

kakao / buffalo

TOROS Buffalo: A fast and scalable production-ready open source project for recommender systems
Apache License 2.0
575 stars 106 forks source link

Multi-GPU Support #35

Closed tangzhy closed 1 year ago

tangzhy commented 3 years ago

Do you support training models among multiple gpus?

I'm asking this because my sparse matrix is quite large like:

ummae commented 3 years ago

@tangzhy It would be helpful to deal large scale data but, we don't have plan to support multiple gpus yet. did you try buffalo on the database you mentioned? I wonder how long does it take to finish training.