Enum rust_bert::pipelines::common::ConfigOption

source · [−]

pub enum ConfigOption {
Show 22 variants
    Bart(BartConfig),
    Bert(BertConfig),
    DistilBert(DistilBertConfig),
    Deberta(DebertaConfig),
    DebertaV2(DebertaV2Config),
    Electra(ElectraConfig),
    Marian(MarianConfig),
    MobileBert(MobileBertConfig),
    OpenAiGpt(OpenAiGptConfig),
    T5(T5Config),
    Albert(AlbertConfig),
    XLNet(XLNetConfig),
    GPT2(Gpt2Config),
    Reformer(ReformerConfig),
    Roberta(RobertaConfig),
    ProphetNet(ProphetNetConfig),
    Longformer(LongformerConfig),
    Pegasus(PegasusConfig),
    GPTNeo(GptNeoConfig),
    MBart(MBartConfig),
    M2M100(M2M100Config),
    FNet(FNetConfig),
}

Expand description

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

Variants

`Bart(BartConfig)`

Bart configuration

`Bert(BertConfig)`

Bert configuration

`DistilBert(DistilBertConfig)`

DistilBert configuration

`Deberta(DebertaConfig)`

DeBERTa configuration

`DebertaV2(DebertaV2Config)`

DeBERTa V2 configuration

`Electra(ElectraConfig)`

Electra configuration

`Marian(MarianConfig)`

Marian configuration

`MobileBert(MobileBertConfig)`

MobileBert configuration

`OpenAiGpt(OpenAiGptConfig)`

OpenAI GPT configuration

`T5(T5Config)`

T5 configuration

`Albert(AlbertConfig)`

Albert configuration

`XLNet(XLNetConfig)`

XLNet configuration

`GPT2(Gpt2Config)`

GPT2 configuration

`Reformer(ReformerConfig)`

Reformer configuration

`Roberta(RobertaConfig)`

RoBERTa configuration

`ProphetNet(ProphetNetConfig)`

ProphetNet configuration

`Longformer(LongformerConfig)`

Longformer configuration

`Pegasus(PegasusConfig)`

Pegasus configuration

`GPTNeo(GptNeoConfig)`

GPT-Neo configuration

`MBart(MBartConfig)`

MBart configuration

`M2M100(M2M100Config)`

M2M100 configuration

`FNet(FNetConfig)`

FNet configuration

Implementations

impl ConfigOption

pub fn from_file<P: AsRef<Path>>(model_type: ModelType, path: P) -> Self

Interface method to load a configuration from file

pub fn get_label_mapping(&self) -> &HashMap<i64, String>

pub fn get_max_len(&self) -> Option<i64>

Trait Implementations

impl TryFrom<&ConfigOption> for AlbertConfig

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<&ConfigOption> for BertConfig

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<&ConfigOption> for DistilBertConfig

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

impl TryFrom<&ConfigOption> for T5Config

type Error = RustBertError

The type returned in the event of a conversion error.

fn try_from(config: &ConfigOption) -> Result<Self, Self::Error>

Performs the conversion.

Auto Trait Implementations

impl RefUnwindSafe for ConfigOption

impl Send for ConfigOption

impl Sync for ConfigOption

impl Unpin for ConfigOption

impl UnwindSafe for ConfigOption

Blanket Implementations

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

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T where
T: ?Sized,

const: unstable · source

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T where
T: ?Sized,

const: unstable · source

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

const: unstable · source

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided Span, returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

impl<T, U> Into<U> for T where
U: From<T>,

const: unstable · source

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Pointable for T

const ALIGN: usize = mem::align_of::<T>()

The alignment of pointer.

type Init = T

The type for initializers.

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more

impl<T> Same<T> for T

type Output = T

Should always be Self

impl<T, U> TryFrom<U> for T where
U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

const: unstable · source

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T where
U: TryFrom<T>,

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

The type returned in the event of a conversion error.

const: unstable · source

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

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

fn vzip(self) -> V