Forráskód Böngészése

换用github jwjohns/unsloth-GRPO-qwen2.5 验证GRPO训练模型

zhouyang.xie 2 hónapja
szülő
commit
5f6552665d
1 módosított fájl, 1 hozzáadás és 1 törlés
  1. 1 1
      src/train_model_github_jwjohns.py

+ 1 - 1
src/train_model_github_jwjohns.py

@@ -75,7 +75,7 @@ from modelscope.msdatasets import MsDataset
 def get_gsm8k_questions(split="train") -> Dataset:
     # data = load_dataset('openai/gsm8k', 'main')[split]
     data =  MsDataset.load('openai-mirror/gsm8k', subset_name='main', split=split)
-    os.makedirs(f'../data/temp/',exist_ok=False)
+    os.makedirs(f'../data/temp/',exist_ok=True)
     # Save original datasets to JSONL
     with open(f'../data/temp/gsm8k_original_{split}.jsonl', 'w') as f:
         for item in data: