Struct LayerFreezingConfig 

pub struct LayerFreezingConfig { /* private fields */ }

Layer freezing configuration for transfer learning.

Implementations

impl LayerFreezingConfig

pub fn new() -> Self

Create a new layer freezing configuration.

pub fn freeze_layers(&mut self, layer_names: &[&str])

Freeze specific layers.

Arguments

• layer_names - Names of layers to freeze

pub fn unfreeze_layers(&mut self, layer_names: &[&str])

Unfreeze specific layers.

Arguments

• layer_names - Names of layers to unfreeze

pub fn freeze_all(&mut self)

Freeze all layers.

pub fn unfreeze_all(&mut self)

Unfreeze all layers.

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

Check if a layer is frozen.

Arguments

• layer_name - Name of the layer to check

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

Get all frozen layer names.

pub fn num_frozen(&self) -> usize

Get the number of frozen layers.

Trait Implementations

impl Clone for LayerFreezingConfig

fn clone(&self) -> LayerFreezingConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LayerFreezingConfig

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

Formats the value using the given formatter.

impl Default for LayerFreezingConfig

fn default() -> Self

Returns the “default value” for a type.