Struct BertModel 

pub struct BertModel { /* private fields */ }

Implementations

impl BertModel

pub fn new(config: BertConfig) -> Result<Self>

pub fn new_with_device(config: BertConfig, device: Device) -> Result<Self>

pub fn device(&self) -> Device

pub fn forward_with_embeddings( &self, input_ids: Vec<u32>, attention_mask: Option<Vec<u8>>, token_type_ids: Option<Vec<u32>>, ) -> Result<BertModelOutput>

Trait Implementations

impl Clone for BertModel

fn clone(&self) -> BertModel

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BertModel

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

Formats the value using the given formatter.

impl Model for BertModel

type Config = BertConfig

type Input = TokenizedInput

type Output = BertModelOutput

fn forward(&self, input: Self::Input) -> Result<Self::Output>

Performs a forward pass through the model.

fn load_pretrained(&mut self, reader: &mut dyn Read) -> Result<()>

Loads pretrained weights into the model.

fn get_config(&self) -> &<BertModel as Model>::Config

Returns a reference to the model’s configuration.

fn num_parameters(&self) -> usize

Returns the total number of parameters in the model.