integrations.kd.callbacks

integrations.kd.callbacks

Transformers trainer callbacks to schedule the KD temperature during training

Classes

Name	Description
KDTemperatureSchedulerCallback	KD temperature scheduler callback for the trainer.

KDTemperatureSchedulerCallback

integrations.kd.callbacks.KDTemperatureSchedulerCallback(
    temperature_start,
    temperature_min,
    trainer,
)

KD temperature scheduler callback for the trainer.