Struct ProgressiveUnfreezing 

pub struct ProgressiveUnfreezing { /* private fields */ }

Progressive unfreezing strategy for transfer learning.

Gradually unfreezes layers from top to bottom during training.

Implementations

impl ProgressiveUnfreezing

pub fn new( layer_order: Vec<String>, unfreeze_interval: usize, ) -> TrainResult<Self>

Create a new progressive unfreezing strategy.

Arguments

• layer_order - Layer names ordered from bottom to top
• unfreeze_interval - Epochs between unfreezing stages

pub fn update_stage(&mut self, epoch: usize) -> bool

Update the unfreezing stage based on current epoch.

Arguments

• epoch - Current training epoch

Returns

Whether the stage was updated

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

Get layers that should be unfrozen at current stage.

Returns

Layer names that should be trainable

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

Get layers that should be frozen at current stage.

Returns

Layer names that should be frozen

pub fn is_complete(&self) -> bool

Check if unfreezing is complete.

pub fn current_stage(&self) -> usize

Get current stage number.

pub fn total_stages(&self) -> usize

Get total number of stages.

Trait Implementations

impl Clone for ProgressiveUnfreezing

fn clone(&self) -> ProgressiveUnfreezing

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ProgressiveUnfreezing

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

Formats the value using the given formatter.