HFDataset#

class stable_pretraining.data.HFDataset(*args, transform=None, rename_columns=None, remove_columns=None, **kwargs)[source]#

Bases: Dataset

Hugging Face dataset wrapper with transform and column manipulation support.

Examples using HFDataset:#

<no title>

Multi-layer probe for vision models.

Supervised Learning Example

Supervised Learning Example