Struct RobertaEmbeddingsConfig

pub struct RobertaEmbeddingsConfig { /* private fields */ }

RoBERTa embeddings configuration.

Implementations

impl RobertaEmbeddingsConfig

pub fn padding_id(self, padding_id: u32) -> Self

Padding token id.

Default: 1

pub fn transformer_embeddings( self, transformer_embeddings: Box<dyn BuildEmbeddings>, ) -> Self

Transformer embeddings configuration.

Default: TransformerEmbeddingsConfig::default()

Trait Implementations

impl BuildEmbeddings for RobertaEmbeddingsConfig

fn build(&self, vb: VarBuilder<'_>) -> Result<Box<dyn Embeddings>, BoxedError>

impl Debug for RobertaEmbeddingsConfig

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

Formats the value using the given formatter.

impl Default for RobertaEmbeddingsConfig

fn default() -> Self

Returns the “default value” for a type.