search

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

fix is_classification error for adversarial training regressor and classification #131

Closed SarthakJariwala closed 1 year ago

SarthakJariwala commented 1 year ago

Fixes #130

xuyxu commented 1 year ago

Merged, thanks!