SimpleTuner icon indicating copy to clipboard operation
SimpleTuner copied to clipboard

[Feature Request] SwanLab Integration as Logging Tracker

Open Puiching-Memory opened this issue 1 month ago • 1 comments

Summary

SwanLab is a wonderful open-source experiment tracking tool, which grants users enhanced flexibility in their choice of logging platforms.

Technical Analysis

https://docs.swanlab.cn/guide_cloud/integration/integration-huggingface-accelerate.html

Accelerate has already integrated SwanLab, so adding an extra logging option shouldn't be overly complex. Similar to wandb, it requires some initialization parameters.

from accelerate import Accelerator
accelerator = Accelerator(log_with="swanlab")
accelerator.init_trackers(
    ...
    init_kwargs={"swanlab": {"experiment_name": "hello_world"}}
    )

Thank you for considering this feature request! I'm happy to help test or provide additional information if needed.

Puiching-Memory avatar Nov 25 '25 11:11 Puiching-Memory

i think a pull request for this one would be as simple as extending the options available to --report_to if you would like to open one

bghira avatar Nov 25 '25 12:11 bghira

I don't have a setup for SwanLab to test: https://github.com/bghira/SimpleTuner/pull/2063

bghira avatar Dec 03 '25 16:12 bghira