Struct MlpBurnConfig 

pub struct MlpBurnConfig {
    pub num_layers: usize,
    pub hidden_dim: usize,
    pub use_orthogonal_init: bool,
    pub activation: BurnActivation,
    pub seed: Option<u64>,
}

Configuration for MlpBurnPolicy architecture.

Mirrors crate::policy::mlp::MlpBurnConfig on the tch path. Stored inside the policy so the parity tests can compare both backends on identical hyperparameters.

Fields

num_layers: usize

Number of hidden layers in the shared trunk. Only 2 or 3 are supported; anything else is treated as 2.

hidden_dim: usize

Width of every hidden layer.

use_orthogonal_init: bool

If true, initialize hidden-layer weights with Initializer::Orthogonal (gain sqrt(2)) and output heads with Initializer::Orthogonal { gain = 0.01 }. Set false to fall back to Burn’s default Kaiming-uniform init.

activation: BurnActivation

Activation applied between hidden layers.

seed: Option<u64>

Optional construction seed. When Some, with_config builds every layer from a deterministic, StdRng- driven weight buffer (see crate::policy::seeded_init) instead of Burn’s unseedable Initializer, so two constructions with the same seed produce bit-identical policies. When None (the default) the behavior is unchanged — Burn’s Initializer path is used verbatim. This is the load-bearing knob behind the end-to-end PsroConfig::seed / NfspConfig::seed reproducibility contract (issue #135). The seeded path covers both the orthogonal and Kaiming-uniform recipes (selected by use_orthogonal_init), so seeding works regardless of which init the caller picks.

Implementations

impl MlpBurnConfig

pub fn with_seed(self, seed: u64) -> Self

Set the construction seed, enabling the deterministic host-side init path in with_config.

Builder-style; returns self for chaining: MlpBurnConfig::default().with_seed(42).

Trait Implementations

impl Clone for MlpBurnConfig

fn clone(&self) -> MlpBurnConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Copy for MlpBurnConfig

impl Debug for MlpBurnConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for MlpBurnConfig

fn default() -> Self

Returns the “default value” for a type.