kashif HF staff commited on
Commit
dfc4ae0
1 Parent(s): 63b10e4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -5,7 +5,8 @@
5
  "InformerForPrediction"
6
  ],
7
  "attention_dropout": 0.1,
8
- "attn": "prob",
 
9
  "cardinality": [
10
  366
11
  ],
@@ -25,7 +26,6 @@
25
  "encoder_ffn_dim": 32,
26
  "encoder_layerdrop": 0.1,
27
  "encoder_layers": 4,
28
- "factor": 2,
29
  "init_std": 0.02,
30
  "input_size": 1,
31
  "is_encoder_decoder": true,
 
5
  "InformerForPrediction"
6
  ],
7
  "attention_dropout": 0.1,
8
+ "attention_factor": 2,
9
+ "attention_type": "prob",
10
  "cardinality": [
11
  366
12
  ],
 
26
  "encoder_ffn_dim": 32,
27
  "encoder_layerdrop": 0.1,
28
  "encoder_layers": 4,
 
29
  "init_std": 0.02,
30
  "input_size": 1,
31
  "is_encoder_decoder": true,