Enum OutputMode 

pub enum OutputMode {
    Classification {
        num_classes: usize,
    },
    Features,
}

Which forward tail the model executes. Switch with [ResNet::with_output] or set at construction time.

Variants

Classification

Add the FC head; forward returns logits [B, num_classes]. The FC weight / bias tensors are loaded from fc.weight / fc.bias.

Fields

num_classes: usize

Features

Stop after stage 4; forward returns the final feature map [B, 512 * expansion, H/32, W/32]. The FC weights are not loaded.

Trait Implementations

impl Clone for OutputMode

fn clone(&self) -> OutputMode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for OutputMode

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Copy for OutputMode