fix: copy collate_fn from the original dataloader (#124)

FedericoV · web-flow · commit 7338a4fe3a26 · 2022-08-04T22:53:51.000+08:00
The bagging dataloaders do not copy the collate_fn from the original dataloaders.
diff --git a/torchensemble/bagging.py b/torchensemble/bagging.py
@@ -468,6 +468,7 @@ def _get_bagging_dataloaders(original_dataloader, n_estimators):
             sub_dataset,
             batch_size=original_dataloader.batch_size,
             num_workers=original_dataloader.num_workers,
+            collate_fn=original_dataloader.collate_fn,
             shuffle=True,
         )
         dataloaders.append(dataloader)

Original file line number	Diff line number	Diff line change
`@@ -468,6 +468,7 @@ def _get_bagging_dataloaders(original_dataloader, n_estimators):`
`468`	`468`	`sub_dataset,`
`469`	`469`	`batch_size=original_dataloader.batch_size,`
`470`	`470`	`num_workers=original_dataloader.num_workers,`
	`471`	`+ collate_fn=original_dataloader.collate_fn,`
`471`	`472`	`shuffle=True,`
`472`	`473`	`)`
`473`	`474`	`dataloaders.append(dataloader)`