serpent / config.yaml
kfoughali's picture
Update config.yaml
5c38da3 verified
raw
history blame
325 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: "bfs" # bfs, spectral, degree, community
preserve_locality: true