feat: support dataloader with multiple inputs

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

Closed xuyxu closed 3 years ago

xuyxu commented 3 years ago

resolves #75

[x] code: make existing ensembles adapt to multiple inputs using non-keyword arguments
[x] test: add unit tests on the multiple inputs case for all ensembles