integrations.swanlab.plugins

integrations.swanlab.plugins

SwanLab Plugin for Axolotl

Classes

Name	Description
SwanLabPlugin	SwanLab integration plugin for Axolotl.

integrations.swanlab.plugins.SwanLabPlugin()

SwanLab integration plugin for Axolotl.

Provides experiment tracking, visualization, and logging capabilities using SwanLab (https://swanlab.cn).

Usage in config.yaml: plugins: - axolotl.integrations.swanlab.SwanLabPlugin

use_swanlab: true
swanlab_project: my-project
swanlab_experiment_name: my-experiment
swanlab_mode: cloud  # or 'local', 'offline', 'disabled'

Name	Description
add_callbacks_pre_trainer	Add SwanLab callbacks before trainer creation.
get_input_args	Returns the configuration model for SwanLab integration.
post_trainer_create	Post-trainer creation hook.
pre_model_load	Initialize SwanLab before model loading with runtime checks.
register	Register SwanLab plugin with configuration and conflict detection.

integrations.swanlab.plugins.SwanLabPlugin.add_callbacks_pre_trainer(cfg, model)

Add SwanLab callbacks before trainer creation.

integrations.swanlab.plugins.SwanLabPlugin.get_input_args()

Returns the configuration model for SwanLab integration.

integrations.swanlab.plugins.SwanLabPlugin.post_trainer_create(cfg, trainer)

Post-trainer creation hook.

integrations.swanlab.plugins.SwanLabPlugin.pre_model_load(cfg)

Initialize SwanLab before model loading with runtime checks.

integrations.swanlab.plugins.SwanLabPlugin.register(cfg)