pub fn build_resnet(
graph: &mut Graph,
config: &ArchitectureConfig,
) -> Result<SequentialModel, ModelError>Expand description
Builds a ResNet-family model: stem + residual stages + global-avg-pool + linear head.
pub fn build_resnet(
graph: &mut Graph,
config: &ArchitectureConfig,
) -> Result<SequentialModel, ModelError>Builds a ResNet-family model: stem + residual stages + global-avg-pool + linear head.