Struct ModuleConfig 

pub struct ModuleConfig {
    pub training: bool,
    pub device: DeviceType,
    pub bias: bool,
    pub dropout: f32,
    pub custom: HashMap<String, String>,
}

Generic module configuration

This provides a standard configuration interface that can be extended by specific module types.

Fields

training: bool

Training mode

device: DeviceType

Target device

bias: bool

Whether to use bias terms

dropout: f32

Dropout probability

custom: HashMap<String, String>

Custom parameters (placeholder when serialize feature is disabled)

Implementations

impl ModuleConfig

pub fn new() -> Self

Create a new configuration with default values

pub fn training(self, training: bool) -> Self

Set training mode

pub fn device(self, device: DeviceType) -> Self

Set device

pub fn bias(self, bias: bool) -> Self

Set bias usage

pub fn dropout(self, dropout: f32) -> Self

Set dropout probability

pub fn custom_param<T: Display>(self, name: &str, value: T) -> Self

Add a custom parameter (simplified version when serialize feature is disabled)

pub fn get_custom(&self, name: &str) -> Option<String>

Get a custom parameter (simplified version when serialize feature is disabled)

Trait Implementations

impl Clone for ModuleConfig

fn clone(&self) -> ModuleConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModuleConfig

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

Formats the value using the given formatter.

impl Default for ModuleConfig

fn default() -> Self

Returns the “default value” for a type.