File size: 447 Bytes
7960e3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
{
"architectures": [
"InformerForPrediction"
],
"d_model": 512,
"n_heads": 8,
"e_layers": 3,
"d_layers": 2,
"d_ff": 2048,
"factor": 5,
"dropout": 0.05,
"attn": "prob",
"embed": "fixed",
"activation": "gelu",
"output_attention": false,
"distil": true,
"model_type": "informer",
"prediction_length": 96,
"context_length": 720,
"lradj": "type1",
"torch_dtype": "float16",
"transformers_version": "4.36.0"
} |