Struct PreFabConfig 

pub struct PreFabConfig<C>
where
    C: 'static + Config + Debug + Clone,
{
    pub name: String,
    pub description: String,
    pub builder: Arc<dyn Fn() -> C + Send + Sync>,
    pub weights: Option<PretrainedWeightsMap>,
}

A Config Well-Known Pre-Fab.

Fields

name: String

Name of the model config pre-fab.

description: String

Description of the model config pre-fab.

builder: Arc<dyn Fn() -> C + Send + Sync>

Builder function for the config.

weights: Option<PretrainedWeightsMap>

Pretrained weights map.

Implementations

impl<C> PreFabConfig<C>

where C: 'static + Config + Debug + Clone,

pub fn to_config(&self) -> C

Build a new config.

pub fn lookup_pretrained_weights( &self, name: &str, ) -> Option<PretrainedWeightsDescriptor>

Lookup a descriptor.

pub fn try_lookup_pretrained_weights( &self, name: &str, ) -> Result<PretrainedWeightsDescriptor>

Lookup a descriptor.

pub fn expect_lookup_pretrained_weights( &self, name: &str, ) -> PretrainedWeightsDescriptor

Lookup a descriptor.

Trait Implementations

impl<C> Clone for PreFabConfig<C>

where C: 'static + Config + Debug + Clone + Clone,

fn clone(&self) -> PreFabConfig<C>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<C> Debug for PreFabConfig<C>

where C: 'static + Config + Debug + Clone,

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

Formats the value using the given formatter.

impl<C> From<&StaticPreFabConfig<C>> for PreFabConfig<C>

where C: 'static + Config + Debug + Clone,

fn from(config: &StaticPreFabConfig<C>) -> Self

Converts to this type from the input type.