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

fix readme example #114

Closed Arcify closed 2 years ago

Arcify commented 2 years ago

ensemble.predict takes a tensor and returns the output feature, therefore it should be ensemble.evaluate, which takes a testloader and returns the accuracy.

xuyxu commented 2 years ago

Thanks @Arcify

xuyxu commented 2 years ago

@all-contributors please add @Arcify for doc

allcontributors[bot] commented 2 years ago

@xuyxu

I've put up a pull request to add @Arcify! :tada: