vblagoje commited on
Commit
e836297
1 Parent(s): 9076a33

Update config.json with proper names

Browse files
Files changed (1) hide show
  1. config.json +4 -17
config.json CHANGED
@@ -18,23 +18,10 @@
18
  "pad_token_id": 1,
19
  "type_vocab_size": 1,
20
  "vocab_size": 50265,
21
- "k": 5,
22
- "n_ntype": 4,
23
- "n_etype": 38,
24
- "n_concept": 799273,
25
  "concept_dim": 200,
26
  "gnn_hidden_size": 200,
27
- "concept_in_dim": 1024,
28
- "n_attention_head": 2,
29
- "fc_dim": 200,
30
- "n_fc_layer": 0,
31
- "p_emb": 0.2,
32
- "p_gnn": 0.2,
33
- "p_fc": 0.2,
34
- "init_range":0.02,
35
- "ie_dim":400,
36
- "info_exchange": true,
37
- "ie_layer_num":1,
38
- "sep_ie_layers": false,
39
- "layer_id":-1
40
  }
 
18
  "pad_token_id": 1,
19
  "type_vocab_size": 1,
20
  "vocab_size": 50265,
21
+ "num_gnn_layers": 5,
22
+ "num_node_types": 4,
23
+ "num_edge_types": 38,
 
24
  "concept_dim": 200,
25
  "gnn_hidden_size": 200,
26
+ "ie_dim":400
 
 
 
 
 
 
 
 
 
 
 
 
27
  }