integrations.kd
integrations.kd
Plugin init to add KD support to Axolotl.
Classes
| Name | Description |
|---|---|
| KDPlugin | Plugin for KD support in Axolotl. |
KDPlugin
integrations.kd.KDPlugin()Plugin for KD support in Axolotl.
Methods
| Name | Description |
|---|---|
| add_callbacks_post_trainer | Adds temp scheduler callback to the Trainer instance. |
add_callbacks_post_trainer
integrations.kd.KDPlugin.add_callbacks_post_trainer(cfg, trainer)Adds temp scheduler callback to the Trainer instance.
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| cfg | Any | Configuration object containing the sparse recipe. | required |
| trainer | Trainer | Huggingface Trainer instance. | required |
Returns
| Name | Type | Description |
|---|---|---|
| list | list | List containing the configured callback instances. |