qqaatw commited on
Commit
12cdd1c
1 Parent(s): bd3ee5c

Update model

Browse files
Files changed (2) hide show
  1. config.json +10 -2
  2. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "architectures": [
3
- "RealmEncoder"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "bos_token_id": 0,
@@ -12,14 +12,22 @@
12
  "intermediate_size": 3072,
13
  "layer_norm_eps": 1e-12,
14
  "max_position_embeddings": 512,
 
15
  "model_type": "realm",
16
  "num_attention_heads": 12,
 
17
  "num_candidates": 8,
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 1,
 
 
 
20
  "retriever_proj_size": 128,
 
 
 
21
  "torch_dtype": "float32",
22
- "transformers_version": "4.10.0.dev0",
23
  "type_vocab_size": 2,
24
  "use_cache": true,
25
  "vocab_size": 30522
 
1
  {
2
  "architectures": [
3
+ "RealmKnowledgeAugEncoder"
4
  ],
5
  "attention_probs_dropout_prob": 0.1,
6
  "bos_token_id": 0,
 
12
  "intermediate_size": 3072,
13
  "layer_norm_eps": 1e-12,
14
  "max_position_embeddings": 512,
15
+ "max_span_width": 10,
16
  "model_type": "realm",
17
  "num_attention_heads": 12,
18
+ "num_block_records": 13353718,
19
  "num_candidates": 8,
20
  "num_hidden_layers": 12,
21
  "pad_token_id": 1,
22
+ "reader_beam_size": 5,
23
+ "reader_layer_norm_eps": 0.001,
24
+ "reader_seq_len": 320,
25
  "retriever_proj_size": 128,
26
+ "searcher_beam_size": 5000,
27
+ "searcher_seq_len": 64,
28
+ "span_hidden_size": 256,
29
  "torch_dtype": "float32",
30
+ "transformers_version": "4.16.0.dev0",
31
  "type_vocab_size": 2,
32
  "use_cache": true,
33
  "vocab_size": 30522
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fb1f15cef0b6f17c777c54fc99b750465797dc33aeff30abe0d81fa1a11b3850
3
- size 440502247
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b2c96e2f14262ab704424fdf3a71142bd3bc5d1e67da31fe2f1f80dbfc96673
3
+ size 440502631