fix(train_page): 修正模型训练保存路径

2025-04-14 16:31:00 +08:00 · 2025-04-14 16:31:00 +08:00 · 11a3039775
commit 11a3039775
parent a4289815ba
1 changed files with 1 additions and 1 deletions
--- 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)