[−][src]Enum rust_bert::pipelines::common::ConfigOption

pub enum ConfigOption {
    Bert(BertConfig),
    DistilBert(DistilBertConfig),
    Electra(ElectraConfig),
    Marian(BartConfig),
    T5(T5Config),
    Albert(AlbertConfig),
}

Abstraction that holds a model configuration, can be of any of the supported models

Variants

Bert configuration

DistilBert(DistilBertConfig)

DistilBert configuration

Electra(ElectraConfig)

Electra configuration

Marian(BartConfig)

Marian configuration

T5 configuration

Albert(AlbertConfig)

Albert configuration

Implementations

impl ConfigOption[src]

pub fn from_file(model_type: ModelType, path: &Path) -> Self[src]

Interface method to load a configuration from file

pub fn get_label_mapping(self) -> HashMap<i64, String>[src]

Auto Trait Implementations

Blanket Implementations

impl<T> Any for T where
    T: 'static + ?Sized, [src]

impl<T> Borrow<T> for T where
    T: ?Sized, [src]

impl<T> BorrowMut<T> for T where
    T: ?Sized, [src]

impl<T> From<T> for T[src]

impl<T, U> Into<U> for T where
    U: From<T>, [src]

impl<T, U> TryFrom<U> for T where
    U: Into<T>, [src]

type Error = Infallible

The type returned in the event of a conversion error.

impl<T, U> TryInto<U> for T where
    U: TryFrom<T>, [src]

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

impl<V, T> VZip<V> for T where
    V: MultiLane<T>,