Function build_resnet_custom 

pub fn build_resnet_custom(
    graph: &mut Graph,
    config: &ArchitectureConfig,
) -> Result<SequentialModel, ModelError>

Builds a ResNet with per-stage block counts (bypasses the single-count helper).