cerebras.modelzoo.data.nlp.dpo.DPOSyntheticDataset

Processors for synthetic data for DPO Training

Classes

DPODataset	A class representing a DPODataset inheriting torch.utils.data.Dataset.
DPOSyntheticDataProcessor	Synthetic dataset generator.

previous

cerebras.modelzoo.data.nlp.dpo

next

cerebras.modelzoo.data.nlp.dpo.DPOSyntheticDataset.DPODataset