Struct ResNet 

pub struct ResNet {
    pub config: ResNetConfig,
    /* private fields */
}

Image classification / feature backbone. Construct via one of the loaders (ResNet::from_hub, ResNet::from_safetensors, or ResNet::from_state_dict) — the empty-tensor placeholders in the layer structs are not usable until weights are loaded.

Fields

config: ResNetConfig

Implementations

impl ResNet

pub fn with_zero_weights(config: ResNetConfig) -> Self

Build with all-zero weight placeholders. Used by every loader before a load_state_dict call, and exposed publicly for round-trip tests.

pub fn feature_channels(&self) -> usize

Number of output channels after stage 4 (before any FC head). Useful when consumers want to pre-allocate downstream buffers.

pub fn from_hub( model_id: &str, depth: ResNetDepth, output: OutputMode, ) -> Result<Self>

Download model.safetensors from a HuggingFace Hub repository at the main revision and load it. The repo must publish a flat timm / torchvision-style state dict.

pub fn from_hub_with_revision( model_id: &str, revision: &str, depth: ResNetDepth, output: OutputMode, ) -> Result<Self>

pub fn from_safetensors( path: &Path, depth: ResNetDepth, output: OutputMode, ) -> Result<Self>

Load from a local model.safetensors. The file must use the timm / torchvision key layout (see the module-level docs for the keys).

pub fn from_state_dict(sd: &StateDict, config: ResNetConfig) -> Result<Self>

Build from a preloaded state dict. Runs remap::fold_batchnorm first to translate running_var into invstd and drop num_batches_tracked — the loaded layer structs read directly from the post-fold layout.

pub fn forward(&self, images: &Tensor, batch: &BoundVariable) -> Result<Tensor>

Run the full network on images [max_b, 3, H, W], shrunk to the batch variable’s bound value before the stem. Returns either classification logits [B, num_classes] or the final feature map [B, 512*exp, H/32, W/32], depending on ResNetConfig::output.

Trait Implementations

impl Clone for ResNet

fn clone(&self) -> ResNet

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl HasStateDict for ResNet

fn state_dict(&self, prefix: &str) -> StateDict

fn load_state_dict(&mut self, sd: &StateDict, prefix: &str) -> Result<(), Error>