WithIndexDataset

class torchelie.datasets.WithIndexDataset(ds)

Wrap a dataset. Also returns the index of the accessed element. Original dataset’s attributes are transparently accessible

Parameters

ds (Dataset) – A dataset