Copy collate_fn from the original dataloader.

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 FedericoV closed 2 years ago

FedericoV commented 2 years ago

Very minor fix so that the bagging_dataloaders copy the collate_fn from the original dataloader.

xuyxu commented 2 years ago

Hi @FedericoV, thanks for the PR. I would like to ask, what happens if we dont add the collate_fn param?

xuyxu commented 2 years ago

@all-contributors please add @FedericoV for code

allcontributors[bot] commented 2 years ago

@xuyxu

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