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

pub enum ConfigOption {
Show 18 variants
    Bart(BartConfig),
    Bert(BertConfig),
    DistilBert(DistilBertConfig),
    Electra(ElectraConfig),
    Marian(BartConfig),
    MobileBert(MobileBertConfig),
    T5(T5Config),
    Albert(AlbertConfig),
    XLNet(XLNetConfig),
    GPT2(Gpt2Config),
    Reformer(ReformerConfig),
    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)`

Tuple Fields

0: BartConfig

Bart configuration

`Bert(BertConfig)`

Tuple Fields

0: BertConfig

Bert configuration

`DistilBert(DistilBertConfig)`

Tuple Fields

0: DistilBertConfig

DistilBert configuration

`Electra(ElectraConfig)`

Tuple Fields

0: ElectraConfig

Electra configuration

`Marian(BartConfig)`

Tuple Fields

0: BartConfig

Marian configuration

`MobileBert(MobileBertConfig)`

Tuple Fields

0: MobileBertConfig

MobileBert configuration

`T5(T5Config)`

Tuple Fields

0: T5Config

T5 configuration

`Albert(AlbertConfig)`

Tuple Fields

0: AlbertConfig

Albert configuration

`XLNet(XLNetConfig)`

Tuple Fields

0: XLNetConfig

XLNet configuration

`GPT2(Gpt2Config)`

Tuple Fields

0: Gpt2Config

GPT2 configuration

`Reformer(ReformerConfig)`

Tuple Fields

0: ReformerConfig

Reformer configuration

`ProphetNet(ProphetNetConfig)`

Tuple Fields

0: ProphetNetConfig

ProphetNet configuration

`Longformer(LongformerConfig)`

Tuple Fields

0: LongformerConfig

Longformer configuration

`Pegasus(PegasusConfig)`

Tuple Fields

0: PegasusConfig

Pegasus configuration

`GPTNeo(GptNeoConfig)`

Tuple Fields

0: GptNeoConfig

GPT-Neo configuration

`MBart(MBartConfig)`

Tuple Fields

0: MBartConfig

MBart configuration

`M2M100(M2M100Config)`

Tuple Fields

0: M2M100Config

M2M100 configuration

`FNet(FNetConfig)`

Tuple Fields

0: 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>

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,

pub fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

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

pub fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

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

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

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

pub fn from(t: T) -> T

Performs the conversion.

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>,

pub fn into(self) -> U

Performs the conversion.

impl<T> Pointable for T

pub const ALIGN: usize

The alignment of pointer.

type Init = T

The type for initializers.

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

Initializes a with the given initializer. Read more

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

Dereferences the given pointer. Read more

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

Mutably dereferences the given pointer. Read more

pub 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.

pub 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.

pub 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>,

pub fn vzip(self) -> V