cuml
cuml copied to clipboard
[FEA] Simplify process to train cuml KMeans on GPU and save the model and later load on a CPU machine for inference
I am trying to use KMeans in CUML for fitting the data, but for inference/prediction I want to do it on CPU? Is it possible somehow? I really need a way to predict on CPU. Please help
EDIT:
I feel like it's useful feature for the community, since training and tuning is more resource taking process using GPU make sense but for inference I feel like a CPU machine should do a decent job in production.