From 82e65ccc21cd9dd623568765117de4a07eef6c78 Mon Sep 17 00:00:00 2001
From: ViperEkura <3081035982@qq.com>
Date: Fri, 5 Dec 2025 13:53:50 +0800
Subject: [PATCH] =?UTF-8?q?fix(tools/train):=20=E4=BF=AE=E5=A4=8D=E5=8F=82?=
 =?UTF-8?q?=E6=95=B0=E4=BC=A0=E9=80=92=E9=94=99=E8=AF=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tools/train.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index d4bb15d..8704092 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -56,7 +56,8 @@ def train(
     checkpoint_interval: int,
     checkpoint_dir: str,
     dpo_beta: float,
-    adamw_betas: tuple,
+    adamw_beta1: float,
+    adamw_beta2: float,
     adamw_weight_decay: float,
     max_grad_norm: float,
     embdeding_lr_rate: int,
@@ -112,7 +113,7 @@ def train(
 
     optim = AdamW(
         param_groups,
-        betas=adamw_betas,
+        betas=(adamw_beta1, adamw_beta2),
         weight_decay=adamw_weight_decay
     )