From 42c23798f27977af777587ded7f4845010f0184a Mon Sep 17 00:00:00 2001 From: stephen Date: Wed, 21 Feb 2024 13:47:38 +0800 Subject: [PATCH] update project_kwargs for ppo config --- 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