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

[ENH] Add type checks on the input `estimator` (#4 and #6) #9

Closed xuyxu closed 3 years ago

xuyxu commented 3 years ago

Fix issue #4 and #6