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.07k stars 95 forks source link

Integration in Lightning Flash #84

Open tchaton opened 3 years ago

tchaton commented 3 years ago

Dear people from @Ensemble-Pytorch,

I am currently working on this framework: https://github.com/PyTorchLightning/lightning-flash which is a collection of production ready tasks.

I was wondering if you would be willing to integrate your library into Lightning Flash. There would be some work to make the API very transparent and simpler for the users.

Best, T.C

xuyxu commented 3 years ago

Hi @tchaton, thanks for your attention! I would like to ask that what we should do in order to integrate torchensemble into Lightning Flash.