czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix few-shot rate

This commit is contained in:
czzhangheng 2025-12-29 10:02:03 +08:00
parent 3ae65229c1
commit 7e2fa532de
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
View File

@ -100,9 +100,9 @@ def read_config(config_path):
for x in ["basic", "model", "train"]: for x in ["basic", "model", "train"]:
config[x]["device"] = device config[x]["device"] = device
# few-shot 0.05-0.05-0.9 # few-shot 0.01-0.01-0.98
config["data"]["test_ratio"] = 0.9 config["data"]["test_ratio"] = 0.98
config["data"]["val_ratio"] = 0.05 config["data"]["val_ratio"] = 0.01
return config return config