From 11a3039775b116e72945c624e1cc82759d1b5460 Mon Sep 17 00:00:00 2001 From: carry Date: Mon, 14 Apr 2025 16:31:00 +0800 Subject: [PATCH] =?UTF-8?q?fix(train=5Fpage):=20=E4=BF=AE=E6=AD=A3?= =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E8=AE=AD=E7=BB=83=E4=BF=9D=E5=AD=98=E8=B7=AF?= =?UTF-8?q?=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- frontend/train_page.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frontend/train_page.py b/frontend/train_page.py index e5db3c1..27d307f 100644 --- a/frontend/train_page.py +++ b/frontend/train_page.py @@ -47,7 +47,7 @@ def train_page(): dataset = [ds["message"][0] for ds in dataset["dataset_items"]] train_model(get_model(), get_tokenizer(), - dataset, get_workdir() + "/1", + dataset, get_workdir() + "/training" + "/1", learning_rate, per_device_train_batch_size, epoch, save_steps, lora_rank)