mei2333 commited on
Commit
b5b108d
·
verified ·
1 Parent(s): c5999ba

Upload config/parameter/rho/PEMSD8/rho0.2.json with huggingface_hub

Browse files
config/parameter/rho/PEMSD8/rho0.2.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "PEMSD8-mini",
3
+ "begin_year": 2012,
4
+ "end_year": 2018,
5
+ "dropout": 0.0,
6
+ "lr": 0.01,
7
+ "batch_size": 128,
8
+ "epoch": 60,
9
+ "gpuid": 1,
10
+ "activation": "relu",
11
+ "scheduler": "epo",
12
+ "momentum":0.99,
13
+ "train": 1,
14
+ "y_len": 12,
15
+ "x_len": 12,
16
+ "data_process": 0,
17
+ "auto_test": 1,
18
+ "auto_lr":1,
19
+ "raw_data_path": "data/PEMSD8-mini/finaldata/",
20
+ "save_data_path": "data/PEMSD8-mini/FastData/",
21
+ "graph_path": "data/PEMSD8-mini/graph/",
22
+ "model_path": "res/parameter/rho/PEMSD8/",
23
+ "emb":{
24
+ "input_len":12,
25
+ "input_dim":3,
26
+ "adaptive_emb_dim":32,
27
+ "D^D": 32,
28
+ "D^W":32,
29
+ "D^N":32,
30
+ "output_len":12,
31
+ "num_feat": 1,
32
+ "feature_emb_dim": 128,
33
+ "num_layer":2
34
+ },
35
+ "tcn":{
36
+ "in_channel": 32,
37
+ "out_channel": 32,
38
+ "kernel_size": 3,
39
+ "dilation": 1
40
+ },
41
+ "TMRB":{
42
+ "in_channel":64,
43
+ "out_channel":64,
44
+ "top_k":12,
45
+ "dropout":0.5
46
+ },
47
+ "logname": "rho0.2",
48
+ "strategy": "incremental",
49
+ "init": true,
50
+ "increase": true,
51
+ "num_hops": 1,
52
+
53
+ "replay": true,
54
+ "replay_strategy": "feature",
55
+ "replay_ratio":0.2,
56
+
57
+ "is_TMRB":true,
58
+ "is_update":true,
59
+ "select_k":true
60
+ }