Struct ParameterManager

pub struct ParameterManager<A: Float, D: Dimension> { /* private fields */ }

Neural network parameter manager with lazy registration

Implementations

impl<A: Float + ScalarOperand + Debug, D: Dimension> ParameterManager<A, D>

pub fn new(config: OptimizationConfig<A>) -> Self

Create a new parameter manager

pub fn enable_lazy_mode(&mut self)

Enable lazy registration mode

pub fn disable_lazy_mode(&mut self) -> Result<()>

Disable lazy registration mode and process pending registrations

pub fn register_layer( &mut self, layer_id: LayerId, architecture: LayerArchitecture, )

Register a layer architecture

pub fn set_layer_rule( &mut self, layer_id: LayerId, rule: LayerOptimizationRule<A>, )

Set layer-specific optimization rule

pub fn register_parameter( &mut self, param_id: ParamId, metadata: ParameterMetadata, ) -> Result<()>

Register a parameter

pub fn get_effective_learning_rate(&self, param_id: &ParamId) -> A

Get effective learning rate for a parameter

pub fn get_effective_weight_decay(&self, param_id: &ParamId) -> A

Get effective weight decay for a parameter

pub fn is_parameter_frozen(&self, param_id: &ParamId) -> bool

Check if parameter is frozen

pub fn get_sharing_group(&self, group_name: &str) -> Option<&[ParamId]>

Get parameters in a sharing group

pub fn get_all_parameters(&self) -> &HashMap<ParamId, ParameterMetadata>

Get all registered parameters

pub fn get_layer_architecture( &self, layer_id: &LayerId, ) -> Option<&LayerArchitecture>

Get layer architecture

pub fn get_parameter_metadata( &self, param_id: &ParamId, ) -> Option<&ParameterMetadata>

Get parameter metadata

pub fn update_config(&mut self, config: OptimizationConfig<A>)

Update global configuration

pub fn get_optimizer_state( &self, param_id: &ParamId, ) -> Option<&HashMap<String, Array<A, D>>>

Get optimizer state for parameter

pub fn get_optimizer_state_mut( &mut self, param_id: &ParamId, ) -> Option<&mut HashMap<String, Array<A, D>>>

Get mutable optimizer state for parameter

pub fn init_optimizer_state( &mut self, param_id: &ParamId, state_name: &str, state: Array<A, D>, ) -> Result<()>

Initialize optimizer state for parameter

pub fn reset_optimizer_states(&mut self)

Reset all optimizer states

pub fn get_parameters_by_layer(&self, layer_id: &LayerId) -> Vec<&ParamId>

Get parameters by layer

pub fn get_parameters_by_type(&self, param_type: ParameterType) -> Vec<&ParamId>

Get parameters by type

pub fn get_trainable_parameters(&self) -> Vec<&ParamId>

Get trainable parameters

Trait Implementations

impl<A: Debug + Float, D: Debug + Dimension> Debug for ParameterManager<A, D>

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

Formats the value using the given formatter.