Struct FeatureExtractorMode 

pub struct FeatureExtractorMode {
    pub feature_extractor_name: String,
    pub head_name: String,
}

Feature extraction mode: freeze entire feature extractor.

Only trains the final classification/regression head.

Fields

feature_extractor_name: String

Name of the feature extractor (typically all layers except last).

head_name: String

Name of the head/classifier (typically the last layer).

Implementations

impl FeatureExtractorMode

pub fn new(feature_extractor_name: String, head_name: String) -> Self

Create a new feature extractor mode.

Arguments

• feature_extractor_name - Name/prefix of feature extractor layers
• head_name - Name/prefix of head layers

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

Check if a layer is part of the feature extractor.

Arguments

• layer_name - Name of the layer

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

Check if a layer is part of the head.

Arguments

• layer_name - Name of the layer

pub fn get_freezing_config(&self, all_layers: &[String]) -> LayerFreezingConfig

Get freezing configuration for feature extraction.

Arguments

• all_layers - All layer names in the model

Returns

Layer freezing configuration

Trait Implementations

impl Clone for FeatureExtractorMode

fn clone(&self) -> FeatureExtractorMode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FeatureExtractorMode

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

Formats the value using the given formatter.