Struct TransferLearningManager 

pub struct TransferLearningManager {
    pub freezing_config: LayerFreezingConfig,
    pub progressive_unfreezing: Option<ProgressiveUnfreezing>,
    pub discriminative_finetuning: Option<DiscriminativeFineTuning>,
    /* private fields */
}

Transfer learning strategy manager.

Fields

freezing_config: LayerFreezingConfig

Layer freezing configuration.

progressive_unfreezing: Option<ProgressiveUnfreezing>

Optional progressive unfreezing strategy.

discriminative_finetuning: Option<DiscriminativeFineTuning>

Optional discriminative fine-tuning.

Implementations

impl TransferLearningManager

pub fn new() -> Self

Create a new transfer learning manager.

pub fn with_progressive_unfreezing( self, strategy: ProgressiveUnfreezing, ) -> Self

Set progressive unfreezing strategy.

Arguments

• strategy - Progressive unfreezing configuration

pub fn with_discriminative_finetuning( self, config: DiscriminativeFineTuning, ) -> Self

Set discriminative fine-tuning.

Arguments

• config - Discriminative fine-tuning configuration

pub fn with_feature_extraction( self, mode: FeatureExtractorMode, all_layers: &[String], ) -> Self

Set feature extraction mode.

Arguments

• mode - Feature extraction configuration
• all_layers - All layer names in the model

pub fn on_epoch_begin(&mut self, epoch: usize)

Update for new epoch.

Arguments

• epoch - Current training epoch

pub fn should_update_layer(&self, layer_name: &str) -> bool

Check if a layer should be updated during training.

Arguments

• layer_name - Name of the layer

pub fn get_layer_lr(&self, layer_name: &str, base_lr: f64) -> f64

Get the learning rate for a specific layer.

Arguments

• layer_name - Name of the layer
• base_lr - Base learning rate

Returns

Layer-specific learning rate

pub fn current_epoch(&self) -> usize

Get current epoch.

Trait Implementations

impl Debug for TransferLearningManager

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

Formats the value using the given formatter.

impl Default for TransferLearningManager

fn default() -> Self

Returns the “default value” for a type.