Skip to main content

build_vgg

Function build_vgg 

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

Builds a VGG-style sequential conv network.

Each stage: blocks_per_stage[i] x (Conv3x3 -> BN -> ReLU), then MaxPool2x2.