fix #1026

Former-commit-id: d0940d0dbd03d4bbcc955304566b0d5507edf9e6
2026-02-01 08:13:38 +00:00 · 2023-09-27 22:57:09 +08:00
parent dd623325e8
commit 6c5d8f089e
2 changed files with 10 additions and 1 deletions
--- a/src/llmtuner/tuner/ppo/workflow.py
+++ b/src/llmtuner/tuner/ppo/workflow.py
@@ -42,6 +42,7 @@ def run_ppo(
        ppo_epochs=1,
        max_grad_norm=training_args.max_grad_norm,
        seed=training_args.seed,
+        log_with=training_args.report_to,
        optimize_cuda_cache=True,
        accelerator_kwargs={"step_scheduler_with_optimizer": False}
    )