How to support multi gpus when fitting the ensemble models?

TorchEnsemble-Community / Ensemble-Pytorch

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

https://ensemble-pytorch.readthedocs.io

BSD 3-Clause "New" or "Revised" License

1.05k stars 95 forks source link

How to support multi gpus when fitting the ensemble models? #158

Open TIMEXue opened 8 months ago

TIMEXue commented 8 months ago

I need to solve cuda out of memory..

xuyxu commented 8 months ago

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.