tinker_cookbook.distillation.DistillationDatasetConfig

class tinker_cookbook.distillation.DistillationDatasetConfig()

Configuration for a dataset used in distillation.

Fields:

• dataset_builder ( RLDatasetBuilder)
• teacher_config ( TeacherConfig)
• groups_per_batch ( int)

Referenced by

• tinker_cookbook.distillation.train_on_policy.Config