Views
No views yet
dunzhang/stella_en_1.5B_v5 on a clinical text-pair similarity dataset.1{
2 "train": "./med_sts/clinicalSTS.train.txt",
3 "test": "./med_sts/clinicalSTS.test.txt",
4 "model": [
5 "dunzhang/stella_en_1.5B_v5"
6 ],
7 "epochs": 20,
8 "batch_size": 16,
9 "lr": 2e-05,
10 "weight_decay": 0.01,
11 "warmup_ratio": 0.1,
12 "max_grad_norm": 1.0,
13 "dropout": 0.1,
14 "max_len": 256,
15 "seed": 42,
16 "patience": 5,
17 "output_dir": "./sts_val/clinical-sts-dunzhang-stella_en_1.5B_v5",
18 "wandb_project": "clinical-sts",
19 "hf_repo": "youssefkhalil320/clinical-sts-dunzhang-stella_en_1.5B_v5"
20}0.2526