Dataset

Contents

Dataset#

class stable_pretraining.data.Dataset(transform=None)[source]#

Bases: Dataset

Base dataset class with transform support and PyTorch Lightning integration.