FromTorchDataset

Contents

FromTorchDataset#

class stable_ssl.data.FromTorchDataset(dataset, names, transform=None)[source]#

Bases: Dataset

Wrapper for PyTorch datasets with custom column naming and transforms.