diff --git a/crates/ml-alpha/src/cfc/trunk.rs b/crates/ml-alpha/src/cfc/trunk.rs index 0db95639f..da2c82af6 100644 --- a/crates/ml-alpha/src/cfc/trunk.rs +++ b/crates/ml-alpha/src/cfc/trunk.rs @@ -101,6 +101,11 @@ pub struct CfcConfig { /// (X1) and the per-stack Mamba2Block constructions added in X3/X5. /// Default matches `PerceptionTrainerConfig::default().mamba2_state_dim`. pub mamba2_state_dim: usize, + /// X11: inference batch + sequence dims for the v2 forward buffers + /// allocated on the trunk. Backtesters typically use (1, 32); + /// training-time consumers (PerceptionTrainer) pass their own dims. + pub n_batch: usize, + pub seq_len: usize, } impl Default for CfcConfig { @@ -110,6 +115,8 @@ impl Default for CfcConfig { n_hid: HIDDEN_DIM, cfc_n_in: HIDDEN_DIM, mamba2_state_dim: 16, + n_batch: 1, + seq_len: 32, } } } diff --git a/crates/ml-alpha/src/trainer/perception.rs b/crates/ml-alpha/src/trainer/perception.rs index 5e338b612..d238c3a2c 100644 --- a/crates/ml-alpha/src/trainer/perception.rs +++ b/crates/ml-alpha/src/trainer/perception.rs @@ -515,6 +515,8 @@ impl PerceptionTrainer { n_hid: HIDDEN_DIM, cfc_n_in: HIDDEN_DIM, mamba2_state_dim: cfg.mamba2_state_dim, + n_batch: cfg.n_batch, + seq_len: cfg.seq_len, }, cfg.seed, )