Add `set_optimizer` and `set_scheduler` for all ensembles

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

Closed xuyxu closed 3 years ago

xuyxu commented 3 years ago