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

feat: add internal `unsqueeze` operation in `forward` of all classifiers #136

Closed xuyxu closed 1 year ago

xuyxu commented 1 year ago

resolves #135