-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig_experiment_sample.json
41 lines (41 loc) · 1.03 KB
/
config_experiment_sample.json
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
{
"model_type": "deberta",
"data_format_mode": 0,
"classification_scheme": 2,
"pretrained_model": "microsoft/deberta-base",
"data_dir": "../sample_data",
"new_model_dir": "../deberta_re_model",
"predict_output_file": "../deberta_re_predict.txt",
"overwrite_model_dir": true,
"seed": 1234,
"max_seq_length": 128,
"cache_data": false,
"data_file_header": true,
"do_train": true,
"do_eval": false,
"do_predict": true,
"do_lower_case": true,
"train_batch_size": 2,
"eval_batch_size": 32,
"learning_rate": 1e-05,
"num_train_epochs": 5,
"gradient_accumulation_steps": 1,
"do_warmup": true,
"warmup_ratio": 0.1,
"weight_decay": 0.0,
"adam_epsilon": 1e-08,
"max_grad_norm": 1.0,
"max_num_checkpoints": 0,
"log_file": null,
"log_lvl": "i",
"log_step": 2,
"num_core": 4,
"non_relation_label": "nonRel",
"progress_bar": false,
"fp16": false,
"fp16_opt_level": "O1",
"use_focal_loss": false,
"focal_loss_gamma": 2,
"use_binary_classification_mode": false,
"balance_sample_weights": false
}