From 7e2fa532de1d81f36c6151239fe91ff241bc4dbb Mon Sep 17 00:00:00 2001 From: czzhangheng Date: Mon, 29 Dec 2025 10:02:03 +0800 Subject: [PATCH] fix few-shot rate --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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