idevede commited on
Commit
69ee76b
1 Parent(s): b16b4d7

add config

Browse files
Files changed (1) hide show
  1. config.json +56 -0
config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "TEMPO",
3
+ "apply_aux_loss": true,
4
+ "architectures": [
5
+ "GPT2Model"
6
+ ],
7
+ "description": "TEMPO",
8
+ "model_id": "TEMPO_checkpoints/TEMPO-80M_v1",
9
+ "checkpoints": "./checkpoints/",
10
+ "task_name": "long_term_forecast",
11
+ "prompt": 1,
12
+ "num_nodes": 1,
13
+ "seq_len": 336,
14
+ "pred_len": 96,
15
+ "label_len": 96,
16
+ "decay_fac": 0.5,
17
+ "learning_rate": 0.001,
18
+ "batch_size": 256,
19
+ "num_workers": 0,
20
+ "train_epochs": 10,
21
+ "lradj": "type3",
22
+ "patience": 5,
23
+ "gpt_layers": 6,
24
+ "is_gpt": 1,
25
+ "e_layers": 3,
26
+ "d_model": 768,
27
+ "n_heads": 4,
28
+ "d_ff": 768,
29
+ "dropout": 0.3,
30
+ "enc_in": 7,
31
+ "c_out": 1,
32
+ "patch_size": 16,
33
+ "kernel_size": 25,
34
+ "loss_func": "mse",
35
+ "pretrain": 0,
36
+ "freeze": 1,
37
+ "model": "TEMPO",
38
+ "stride": 8,
39
+ "max_len": -1,
40
+ "hid_dim": 16,
41
+ "tmax": 20,
42
+ "itr": 3,
43
+ "cos": 1,
44
+ "equal": 1,
45
+ "pool": false,
46
+ "no_stl_loss": false,
47
+ "stl_weight": 0.001,
48
+ "config_path": "./configs/custom_dataset.yml",
49
+ "datasets": "ETTm1,ETTh1,ETTm2,electricity,traffic,weather",
50
+ "target_data": "Custom",
51
+ "use_token": 0,
52
+ "electri_multiplier": 1,
53
+ "traffic_multiplier": 1,
54
+ "embed": "timeF",
55
+ "percent": 100
56
+ }