Stoicheia-code / configs /syntax /joint_greberta_f1_s1.json
Ericu950's picture
Stoicheia: training and evaluation code
5952424 verified
Raw
History Blame Contribute Delete
731 Bytes
{
"kfold_dir": "$STOICHEIA_DATA/treebanks/oga_repo/kfold",
"attn": "sdpa",
"T": 2048,
"W": 384,
"micro_batch": 32,
"eval_micro": 64,
"lr_enc": 0.0001,
"lr_head": 0.0008,
"llrd": 1.0,
"wd": 0.01,
"clip": 1.0,
"warmup_frac": 0.1,
"epochs": 60,
"patience": 8,
"pool": "mean",
"head_dropout": 0.33,
"parse_dropout": 0.33,
"w_xpos": 1.0,
"w_script": 1.0,
"w_upos": 0.2,
"w_flat": 1.0,
"use_cap": true,
"scalar_mix": true,
"d_arc": 500,
"d_rel": 150,
"w_parse": 1.0,
"log_every": 10,
"backbone": "hf",
"hf_max_len": 512,
"hf_model_name_or_path": "bowphs/GreBerta",
"name": "joint_greberta_f1_s1",
"seed": 1,
"fold": 1,
"out_dir": "$SYN_DATA/runs/joint_greberta_f1_s1"
}