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

loading an ensemble from checkpoint #41

Closed mttgdd closed 3 years ago

xuyxu commented 3 years ago

Hi @mttgdd, thanks for this useful PR! Could you further check the code quality CI and fix some small problems on the coding style?

mttgdd commented 3 years ago

Of course, no problem. I'll work on those when I get a moment.

xuyxu commented 3 years ago

Great, thanks @mttgdd! After making some modifications to the documentation, I will merge this PR ;-)

xuyxu commented 3 years ago

Merged ;-)