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

pub enum ConfigOption {
    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),
}

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

Electra(ElectraConfig)

Electra configuration

Marian(BartConfig)

Marian configuration

MobileBert(MobileBertConfig)

MobileBert configuration

T5(T5Config)

T5 configuration

Albert(AlbertConfig)

Albert configuration

XLNet(XLNetConfig)

XLNet configuration

GPT2(Gpt2Config)

GPT2 configuration

Reformer(ReformerConfig)

Reformer configuration

ProphetNet(ProphetNetConfig)

ProphetNet configuration

Longformer(LongformerConfig)

Longformer configuration

Implementations

`impl ConfigOption`[src]

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

Interface method to load a configuration from file

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

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,` [src]

`pub fn type_id(&self) -> TypeId`[src]

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

`pub fn borrow(&self) -> &T`[src]

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

`pub fn borrow_mut(&mut self) -> &mut T`[src]

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

`pub fn from(t: T) -> T`[src]

`impl<T> Instrument for T`[src]

`pub fn instrument(self, span: Span) -> Instrumented<Self>`[src]

`pub fn in_current_span(self) -> Instrumented<Self>`[src]

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

`pub fn into(self) -> U`[src]

`impl<T> Pointable for T`

`pub const ALIGN: usize`

`type Init = T`

The type for initializers.

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

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

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

`pub unsafe fn drop(ptr: usize)`

`impl<T> Same<T> for T`

`type Output = T`

Should always be Self

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

`type Error = Infallible`

The type returned in the event of a conversion error.

`pub fn try_from(value: U) -> Result<T, <T as TryFrom>::Error>`[src]

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

`type Error = >::Error`

The type returned in the event of a conversion error.

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

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

Variants

Implementations

impl ConfigOption[src]

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

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

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, [src]

pub fn type_id(&self) -> TypeId[src]

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

pub fn borrow(&self) -> &T[src]

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

pub fn borrow_mut(&mut self) -> &mut T[src]

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

pub fn from(t: T) -> T[src]

impl<T> Instrument for T[src]

pub fn instrument(self, span: Span) -> Instrumented<Self>[src]

pub fn in_current_span(self) -> Instrumented<Self>[src]

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

pub fn into(self) -> U[src]

impl<T> Pointable for T

pub const ALIGN: usize

type Init = T

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

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

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

pub unsafe fn drop(ptr: usize)

impl<T> Same<T> for T

type Output = T

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

type Error = Infallible

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

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

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

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

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

pub fn vzip(self) -> V