HFDataset

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

Bases: Dataset

Hugging Face dataset wrapper with transform and column manipulation support.

Examples using HFDataset:

sphx_glr_auto_examples_multi_layer_probe.py

Multi-layer probe for vision models.

Supervised Learning Example

Supervised Learning Example