Sfoglia il codice sorgente

fixing name to test

Hamid Shojanazeri 1 anno fa
parent
commit
f0f0c156df
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      src/llama_recipes/configs/datasets.py

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

@@ -31,5 +31,5 @@ class custom_dataset:
     dataset: str = "custom_dataset"
     dataset: str = "custom_dataset"
     file: str = "examples/custom_dataset.py"
     file: str = "examples/custom_dataset.py"
     train_split: str = "train"
     train_split: str = "train"
-    test_split: str = "validation"
+    test_split: str = "test"
     data_path: str = "custom_dataset.json"
     data_path: str = "custom_dataset.json"