Ver Fonte

更换unsloth grpo的训练数据集并验证

zhouyang.xie há 4 meses atrás
pai
commit
39bf52bf4e
1 ficheiros alterados com 1 adições e 1 exclusões
  1. 1 1
      src/train_model_grpo.py

+ 1 - 1
src/train_model_grpo.py

@@ -153,7 +153,7 @@ class ModelTrainer:
 
     def load_data(self, train_data_path):
         # 加载训练集和测试集
-        data = load_dataset("json", data_files={"train": train_data_path}, split="train")
+        train_dataset = load_dataset("json", data_files={"train": train_data_path}, split="train")
 
         # train_loader = torch.utils.data.DataLoader(
         #     train_dataset, batch_size=1, shuffle=True, pin_memory=True  # 启用 pin_memory  2025年3月7日未能验证通过