[BUG] Ctrl+C 中断训练后，SwanLab 实验未自动关闭，导致数据丢失风险

[laoshuaib]

🐛 Bug description

使用 SwanLabLogger 集成 PyTorch Lightning 进行训练时，如果用户通过 Ctrl+C 中断训练进程，SwanLab 实验不会被自动关闭。这会导致：

• 实验在 Web UI 中持续显示为“运行中”状态。
• 缓冲区中的日志和指标数据可能丢失，无法上传到云端。
• 资源未被正确释放。

我并不熟悉该领域，但我查看源码感觉或许该问题的根本原因在于 SwanLabLogger 的 finalize 方法实现不完整，且 SwanLab SDK 缺少像 atexit 或信号处理器这样的兜底机制来处理非正常退出。

---

🧑‍💻 Step to reproduce

1. 参考官方文档，使用 SwanLabLogger 初始化一个 PyTorch Lightning Trainer。

   from swanlab.integration.pytorch_lightning import SwanLabLogger
   import pytorch_lightning as pl
   
   logger = SwanLabLogger(project="test-interrupt")
   trainer = pl.Trainer(max_epochs=10, logger=logger) # 设置一个较长的epoch以便中断

2. 启动训练：trainer.fit(model, datamodule)。
3. 在训练过程中，按下 Ctrl+C 中断程序。
4. 登录 SwanLab Web 控制台，查看名为 test-interrupt 的项目。

---

👾 Expected result

• 即使训练被 Ctrl+C 中断，实验也应在 Web UI 中被标记为“已中断”或“失败”，而不是“运行中”。
• 所有在中断前已记录的日志和指标应被完整上传。
• 系统资源应被正确释放。

[SAKURA-CAT]

源码中有使用 atexit ，您这种情况可能是因为有其他库覆盖了 swanlab 注册在 atexit 上的处理器

[laoshuaib]

进一步补充，我发现正常退出后，下次开启实验依旧报警

python3.10/site-packages/swanlab/integration/pytorch_lightning.py:107: There is a swanlab experiment already in progress and newly created instances of `SwanLabLogger` will reuse this experiment. If this is not desired, call `swanlab.finish()` before instantiating `SwanLabLogger`.

我是使用hydra管理配置，hydra的配置项如下：

trainer:
  _target_: lightning.Trainer
  accelerator: auto
  strategy: auto
  devices: auto
  num_nodes: 1
  precision: 32
  logger:
    _target_: swanlab.integration.pytorch_lightning.SwanLabLogger
    experiment_name: abc
    logdir: ${hydra:runtime.output_dir}
    project: ABC

在main函数这样实例化：

@hydra.main(version_base="1.3", config_name="default.yaml")
def main(cfg: DictConfig):
    if cfg.tf32:
        assert cfg.trainer.precision == 32
        torch.backends.cuda.matmul.allow_tf32 = True
        torch.backends.cudnn.allow_tf32 = True

    model: L.LightningModule = instantiate(cfg.model, _convert_="all")

    if cfg.compile:
        model.module.compile(mode=cfg.compile)
    trainer: L.Trainer = instantiate(cfg.trainer)

现在每次启动实验都会报上述警告，我不清楚这是哪里的问题，但从结果来看swanLab的记录是正常的；
难道这样的声明存在重复初始化？我有什么办法可以消除这个警告吗

anyway，这是一个令人困惑的警告，但它不影响正常使用