kfoughali commited on
Commit
e0c1384
·
verified ·
1 Parent(s): 2bc24ad

Create config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +22 -0
config.yaml ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ d_model: 256
3
+ d_state: 16
4
+ d_conv: 4
5
+ expand: 2
6
+ n_layers: 4
7
+ dropout: 0.1
8
+
9
+ data:
10
+ batch_size: 32
11
+ max_nodes: 10000
12
+ test_split: 0.2
13
+
14
+ training:
15
+ learning_rate: 1e-4
16
+ weight_decay: 0.01
17
+ epochs: 100
18
+ patience: 10
19
+
20
+ ordering:
21
+ strategy: "multi_view" # bfs, spectral, degree, multi_view
22
+ preserve_locality: true