integrations.kd.trainer
integrations.kd.trainer
KD trainer
Classes
| Name | Description |
|---|---|
| AxolotlKDTrainer | Custom trainer subclass for Knowledge Distillation (KD) |
AxolotlKDTrainer
integrations.kd.trainer.AxolotlKDTrainer(*args, **kwargs)Custom trainer subclass for Knowledge Distillation (KD)