Using GPU for training (BanditRecommender)

[TengHoo3]

Hi there,

Just want to check whether this package supports GPU training where I can put the BanditRecommender model into GPU to train along with the dataset?

Cheers!

[bkleyn]

Hi @TengHoo3,

Thank you for your question and interest in the library!

Mab2Rec does not support GPU training at present. Since, most multi-armed bandit algorithms do not make use of matrix operations in the same way as Deep Learning algorithms, the benefits from GPU acceleration is not as great. However, there is certainly opportunity to speed up some of the data processing functions using GPU.

Kind regards,