fix few-shot rate

2025-12-29 10:02:03 +08:00 · 2025-12-29 10:02:03 +08:00 · 7e2fa532de
parent 3ae65229c1
commit 7e2fa532de
1 changed files with 3 additions and 3 deletions
--- a/train.py
+++ b/train.py
@ -100,9 +100,9 @@ def read_config(config_path):
    for x in ["basic", "model", "train"]:
        config[x]["device"] = device

-    # few-shot 0.05-0.05-0.9
-    config["data"]["test_ratio"] = 0.9
-    config["data"]["val_ratio"] = 0.05
+    # few-shot 0.01-0.01-0.98
+    config["data"]["test_ratio"] = 0.98
+    config["data"]["val_ratio"] = 0.01

    return config