Enum ModelArchitecture 

pub enum ModelArchitecture {
    ResNet18,
    ResNet34,
    ResNet50,
    ResNet101,
    VGG16,
    VGG19,
    MobileNetV2,
    EfficientNetB0,
    AlexNet,
    ViTTiny,
    ViTBase,
    ViTLarge,
    DeiTTiny,
}

Known model architectures in the zoo.

Variants

ResNet18

ResNet34

ResNet50

ResNet101

VGG16

VGG19

MobileNetV2

EfficientNetB0

AlexNet

ViTTiny

ViTBase

ViTLarge

DeiTTiny

Implementations

impl ModelArchitecture

pub fn config(&self) -> ArchitectureConfig

Returns the canonical configuration for this architecture.

pub fn name(&self) -> &'static str

Returns a filesystem-safe name for this architecture (used for weight files).

pub fn all() -> &'static [ModelArchitecture]

All known architectures.

Trait Implementations

impl Clone for ModelArchitecture

fn clone(&self) -> ModelArchitecture

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ModelArchitecture

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ModelArchitecture

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for ModelArchitecture

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

Formats the value using the given formatter.

impl Hash for ModelArchitecture

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for ModelArchitecture

fn eq(&self, other: &ModelArchitecture) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for ModelArchitecture

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for ModelArchitecture

impl Eq for ModelArchitecture

impl StructuralPartialEq for ModelArchitecture