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

Copy collate_fn from the original dataloader. #124

Closed FedericoV closed 1 year ago

FedericoV commented 1 year ago

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

xuyxu commented 1 year ago

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

xuyxu commented 1 year ago

@all-contributors please add @FedericoV for code

allcontributors[bot] commented 1 year ago

@xuyxu

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