lumamba 0.0.1

LuMamba EEG foundation model — inference in Rust on the RLX runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

{
  "model_type": "lumamba",
  "model": {
    "patch_size": 40,
    "num_queries": 6,
    "embed_dim": 64,
    "num_heads": 2,
    "mlp_ratio": 4.0,
    "num_classes": 0,
    "norm_eps": 1e-5,
    "expand": 2,
    "num_blocks": 2,
    "d_state": 16,
    "d_conv": 4,
    "bidirectional_strategy": "add"
  }
}