Struct PretrainedWeightLoader

pub struct PretrainedWeightLoader { /* private fields */ }

Pre-trained model weight loader

Implementations

impl PretrainedWeightLoader

pub fn new() -> Self

Create a new weight loader

pub fn load_weights( &mut self, weights: HashMap<String, ArrayD<f32>>, ) -> Result<()>

Load weights from a dictionary

pub fn add_layer_mapping(&mut self, source_layer: String, target_layer: String)

Add layer mapping for weight transfer

pub fn set_ignore_mismatches(&mut self, ignore: bool)

Set whether to ignore size mismatches

pub fn apply_weights_to_layer<L: Layer<f32>>( &self, layer: &mut L, layer_name: &str, ) -> Result<bool>

Apply weights to a model layer with shape compatibility checking

pub fn get_available_weights(&self) -> Vec<String>

Get available weight keys

pub fn get_weight_statistics(&self) -> HashMap<String, WeightStatistics>

Get weight statistics for analysis

pub fn check_weight_compatibility( &self, expected_shapes: &HashMap<String, Vec<usize>>, ) -> CompatibilityReport

Check weight compatibility with expected shapes

pub fn load_from_pytorch_state_dict( &mut self, state_dict: HashMap<String, ArrayD<f32>>, ) -> Result<()>

Load weights from various formats

pub fn load_from_tensorflow_checkpoint( &mut self, checkpoint_data: HashMap<String, ArrayD<f32>>, ) -> Result<()>

Load weights from TensorFlow checkpoint format

Trait Implementations

impl Default for PretrainedWeightLoader

fn default() -> Self

Returns the “default value” for a type.