HorizontalConcatDataset

class torchelie.datasets.HorizontalConcatDataset(datasets: List)

Concatenates multiple datasets. However, while torchvision’s ConcatDataset just concatenates samples, torchelie’s also relabels classes. While a vertical concat like torchvision’s is useful to add more examples per class, an horizontal concat merges datasets to more classes.

Parameters

datasets (list of Dataset) – the datasets to concatenate

classes: torchelie.datasets.concat.CatedLists[str]