Function candle_transformers::models::resnet::resnet18
source · pub fn resnet18(num_classes: usize, vb: VarBuilder<'_>) -> Result<Func<'_>>Expand description
Creates a ResNet-18 model.
pub fn resnet18(num_classes: usize, vb: VarBuilder<'_>) -> Result<Func<'_>>Creates a ResNet-18 model.