Nah_kagz1092 commited on
Commit
107d767
·
verified ·
1 Parent(s): e2692aa

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +18 -2
config.json CHANGED
@@ -1,4 +1,20 @@
1
  {
2
- "encoder_layers": 2,
3
- "decoder_layers": 2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  }
 
1
  {
2
+ "encoder_params": {
3
+ "num_layers": 2,
4
+ "d_model": 128,
5
+ "num_heads": 4,
6
+ "dff": 512,
7
+ "input_vocab_size": 8500,
8
+ "maximum_position_encoding": 10000,
9
+ "rate": 0.1
10
+ },
11
+ "decoder_params": {
12
+ "num_layers": 2,
13
+ "d_model": 128,
14
+ "num_heads": 4,
15
+ "dff": 512,
16
+ "target_vocab_size": 8000,
17
+ "maximum_position_encoding": 6000,
18
+ "rate": 0.1
19
+ }
20
  }