[tracking] add swanlab_logdir param (#7219)

* feat: add swanlab_logdir param * fix Former-commit-id: a1e76af3d9
2025-12-29 10:10:35 +08:00 · 2025-03-11 00:53:07 +08:00
parent 5a29f49fb1
commit 0a43bc1960
3 changed files with 31 additions and 4 deletions
--- a/src/llamafactory/train/trainer_utils.py
+++ b/src/llamafactory/train/trainer_utils.py
@@ -617,6 +617,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
        experiment_name=finetuning_args.swanlab_run_name,
        mode=finetuning_args.swanlab_mode,
        config={"Framework": "🦙LlamaFactory"},
+        logdir=finetuning_args.swanlab_logdir,
    )
    return swanlab_callback