Using pretrained models directedly at decision level.

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

Using pretrained models directedly at decision level. #151

Open opencomvis opened 1 year ago

opencomvis commented 1 year ago

Is the a way to use the libaray for pretrained model? suppose that I have 4 alreay trained models, and I want to just ensemble them.. Is there a way to do that ?

xuyxu commented 1 year ago

Hi @opencomvis, if your models are independently trained, i think it would be much easier to use a single loop to average predictions over them :-)