jackboyla commited on
Commit
9d864f1
·
verified ·
1 Parent(s): d779cce

Push model using huggingface_hub.

Browse files
Files changed (2) hide show
  1. glirel_config.json +16 -9
  2. pytorch_model.bin +1 -1
glirel_config.json CHANGED
@@ -1,32 +1,39 @@
1
  {
2
  "lr_encoder": "1e-5",
3
  "lr_others": "1e-4",
4
- "num_steps": 10000,
5
  "warmup_ratio": 0.1,
6
- "train_batch_size": 8,
7
- "eval_every": 500,
8
- "gradient_accumulation": null,
9
  "eval_batch_size": 32,
10
  "max_width": 12,
11
  "model_name": "microsoft/deberta-v3-large",
12
  "fine_tune": true,
13
  "subtoken_pooling": "first",
14
  "hidden_size": 768,
 
15
  "span_mode": "marker",
 
 
 
16
  "dropout": 0.4,
17
  "scheduler": "cosine_with_warmup",
18
  "loss_func": "binary_cross_entropy_loss",
19
- "dataset_name": "few_rel",
20
  "root_dir": "ablation_backbone",
21
- "train_data": "data/few_rel_all.jsonl",
 
 
22
  "prev_path": "none",
23
  "size_sup": -1,
24
  "num_train_rel_types": 25,
25
  "num_unseen_rel_types": 15,
26
  "top_k": 1,
27
- "random_drop": true,
28
  "max_len": 384,
29
- "eval_threshold": 0.1,
30
  "name": "large",
31
- "log_dir": null
 
32
  }
 
1
  {
2
  "lr_encoder": "1e-5",
3
  "lr_others": "1e-4",
4
+ "num_steps": 300000,
5
  "warmup_ratio": 0.1,
6
+ "train_batch_size": 1,
7
+ "eval_every": 3000,
8
+ "gradient_accumulation": 8,
9
  "eval_batch_size": 32,
10
  "max_width": 12,
11
  "model_name": "microsoft/deberta-v3-large",
12
  "fine_tune": true,
13
  "subtoken_pooling": "first",
14
  "hidden_size": 768,
15
+ "scorer": "dot",
16
  "span_mode": "marker",
17
+ "refine_prompt": false,
18
+ "refine_relation": false,
19
+ "ffn_mul": 4,
20
  "dropout": 0.4,
21
  "scheduler": "cosine_with_warmup",
22
  "loss_func": "binary_cross_entropy_loss",
23
+ "dataset_name": "zero_rel",
24
  "root_dir": "ablation_backbone",
25
+ "train_data": [
26
+ "data/zero_rel_all.jsonl"
27
+ ],
28
  "prev_path": "none",
29
  "size_sup": -1,
30
  "num_train_rel_types": 25,
31
  "num_unseen_rel_types": 15,
32
  "top_k": 1,
33
+ "random_drop": false,
34
  "max_len": 384,
35
+ "eval_threshold": 0.001,
36
  "name": "large",
37
+ "log_dir": "logs/zero_rel/zero_rel-2024-06-19__16-34-15",
38
+ "eval_data": "data/wiki_zsl_all.jsonl"
39
  }
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ba1641e51dcc0123bbd26b4ea8b9946fdf5ad2d30c6c048ccc92dbfb069459e8
3
  size 1866500606
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:296483cf26b60796cef031e0c53145826e18536233469fba0a03631f1da471b9
3
  size 1866500606