From 823f618cba9d796febc9c366dd8cc07b47c3163d Mon Sep 17 00:00:00 2001
From: stephen <stephen@gmail.com>
Date: Wed, 21 Feb 2024 13:47:38 +0800
Subject: [PATCH] update project_kwargs for ppo config

Former-commit-id: 14f106962fc0a87802ae9ecffff00d52f7f5f046
---
 src/llmtuner/train/ppo/workflow.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/llmtuner/train/ppo/workflow.py b/src/llmtuner/train/ppo/workflow.py
index 119739d2..50a0e1d0 100644
--- a/src/llmtuner/train/ppo/workflow.py
+++ b/src/llmtuner/train/ppo/workflow.py
@@ -61,6 +61,7 @@ def run_ppo(
         use_score_norm=finetuning_args.ppo_score_norm,
         whiten_rewards=finetuning_args.ppo_whiten_rewards,
         accelerator_kwargs={"step_scheduler_with_optimizer": False},
+        project_kwargs={"logging_dir": training_args.logging_dir},
     )
 
     # Create optimizer and scheduler