Skip to main content

build_feature_extractor

Function build_feature_extractor 

Source
pub fn build_feature_extractor(
    arch: ModelArchitecture,
    graph: &mut Graph,
    config: &ArchitectureConfig,
) -> Result<SequentialModel, ModelError>
Expand description

Builds a backbone (feature extractor) without the final classifier head.