make validation cheaper

Lightning-AI · Apr 25, 2024 · 20e22f2 · 20e22f2
1 parent f08ae19
commit 20e22f2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_readme.py b/tests/test_readme.py
@@ -75,9 +75,10 @@ def test_finetune_model():
     finetune_command = [
         "litgpt", "finetune", "lora",
         "--checkpoint_dir", str(CHECKPOINT_DIR),
+        "--lora_r", "1",
         "--data", "JSON",
         "--data.json_path", str(DATASET_PATH),
-        "--data.val_split_fraction", "0.1",
+        "--data.val_split_fraction", "0.00001", # Keep small because new final validation is expensive
         "--train.max_steps", "1",
         "--out_dir", str(OUT_DIR)
     ]