Skip to content

contrastive

contrastive

contrastive.py Padded contrastive dataset: paired positive/negative sequences with masks.

ContrastivePaddedDataset(spec: ExecutionSpec, block_size: int, name: str, suffix: str = '')

Bases: Dataset