NL2SQL_zero_shot / finetune_config.json
bgglee's picture
VER5 — Zeroshot
3ab84e5 verified
raw
history blame contribute delete
477 Bytes
{
"version": "VER5-Zeroshot",
"base_model": "bgglee/NL2SQL_finetuned",
"merged_model_repo": "bgglee/NL2SQL_zero_shot",
"method": "LoRA (merged full model via CPU)",
"learning_rate": 0.00015,
"max_steps": 80,
"warmup_steps": 10,
"scheduler": "linear",
"batch_size": 4,
"grad_accum": 4,
"lora_r": 8,
"lora_alpha": 16,
"lora_dropout": 0.1,
"max_len": 512,
"dataset": "data/sql2nl_zero_shot_final_100ea.json",
"schema_dir": "data/bird/BIRD_SQLite"
}