Struct LstmBurnConfig 

pub struct LstmBurnConfig {
    pub hidden_dim: usize,
    pub use_orthogonal_init: bool,
    pub seed: Option<u64>,
}

Configuration for LstmBurnPolicy.

Mirrors crate::policy::mlp::MlpBurnConfig’s knobs that carry over to a recurrent trunk. The LSTM has no depth knob (a single recurrent layer for v1) and no activation knob (the gate/cell/hidden activations are fixed by the LSTM cell equations).

Fields

hidden_dim: usize

Width of the LSTM hidden/cell state (and therefore the feature vector fed to both heads).

use_orthogonal_init: bool

If true, use the PPO orthogonal recipe (gain sqrt(2) on the gate weights, 0.01 on the output heads). If false, fall back to Kaiming-uniform (seeded path) or Burn’s default XavierNormal (unseeded path). Only affects the distribution the weights are drawn from, never the reproducibility guarantee.

seed: Option<u64>

Optional construction seed. When Some, with_config builds every gate Linear and both heads from a deterministic, StdRng-driven weight buffer instead of Burn’s unseedable Initializer, so two constructions with the same seed produce bit-identical policies. When None the LSTM uses Burn’s default XavierNormal gate init (unseeded).

Implementations

impl LstmBurnConfig

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

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

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

Trait Implementations

impl Clone for LstmBurnConfig

fn clone(&self) -> LstmBurnConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for LstmBurnConfig

impl Debug for LstmBurnConfig

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

Formats the value using the given formatter.

impl Default for LstmBurnConfig

fn default() -> Self

Returns the “default value” for a type.