From 980299cd548eb8fba92814e1b9e8861d229462f5 Mon Sep 17 00:00:00 2001 From: ViperEkura <3081035982@qq.com> Date: Fri, 20 Mar 2026 21:54:13 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E5=8F=82=E6=95=B0?= =?UTF-8?q?=E4=BC=A0=E9=80=92=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/train.py b/tools/train.py index bf3287f..1186206 100644 --- a/tools/train.py +++ b/tools/train.py @@ -35,7 +35,7 @@ def parse_args() -> argparse.Namespace: parser.add_argument("--window_size", type=int, default=None, help="the max length of the input sequence.") parser.add_argument("--stride", type=int, default=None, help="the step size of the input sequence.") parser.add_argument("--dpo_beta", type=float, default=0.1, help="DPO beta value.") - parser.add_argument("--label_smoothing", type=int, default=0.1, help="cross_entropy function label smoothing parameter") + parser.add_argument("--label_smoothing", type=float, default=0.1, help="cross_entropy function label smoothing parameter") parser.add_argument("--checkpoint_interval", type=int, default=5000, help="Number of iters between checkpoints.") parser.add_argument("--checkpoint_dir", type=str, default="checkpoint", help="Directory to save checkpoints.")