serpent / config.yaml
kfoughali's picture
Create config.yaml
e0c1384 verified
raw
history blame
333 Bytes
model:
d_model: 256
d_state: 16
d_conv: 4
expand: 2
n_layers: 4
dropout: 0.1
data:
batch_size: 32
max_nodes: 10000
test_split: 0.2
training:
learning_rate: 1e-4
weight_decay: 0.01
epochs: 100
patience: 10
ordering:
strategy: "multi_view" # bfs, spectral, degree, multi_view
preserve_locality: true