Ensemble-Pytorch icon indicating copy to clipboard operation
Ensemble-Pytorch copied to clipboard
verified publisher icon TorchEnsemble-Community

How to support multi gpus when fitting the ensemble models?

Open TIMEXue opened this issue 2 years ago • 1 comments

I need to solve cuda out of memory..

TIMEXue avatar Oct 18 '23 05:10 TIMEXue

Hi @TIMEXue, I am sorry to tell you that using multi gpus is not supported for now. New features have not been developed for a while, and a PR is much appreciated.

xuyxu avatar Oct 18 '23 14:10 xuyxu