浏览代码

fix incorrect split of InstructionDataset

exhyy 1 年之前
父节点
当前提交
69db75d425
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/llama_recipes/datasets/alpaca_dataset.py

+ 1 - 1
src/llama_recipes/datasets/alpaca_dataset.py

@@ -27,7 +27,7 @@ class InstructionDataset(Dataset):
     def __init__(self, dataset_config, tokenizer, partition="train"):
         self.ann = json.load(open(dataset_config.data_path))
         if partition == "train":
-            self.ann = self.ann
+            self.ann = self.ann[200:]
         else:
             self.ann = self.ann[:200]