diff --git a/train.py b/train.py
index c78e856..2294713 100644
--- 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