pub struct RetNetForLanguageModeling { /* private fields */ }Expand description
RetNet for language modeling
Implementations§
Source§impl RetNetForLanguageModeling
impl RetNetForLanguageModeling
pub fn new(config: RetNetConfig) -> Result<Self>
pub fn new_with_device(config: RetNetConfig, device: Device) -> Result<Self>
pub fn device(&self) -> Device
Trait Implementations§
Source§impl Model for RetNetForLanguageModeling
impl Model for RetNetForLanguageModeling
type Config = RetNetConfig
type Input = Vec<u32>
type Output = Tensor
Source§fn forward(&self, input: Self::Input) -> Result<Self::Output>
fn forward(&self, input: Self::Input) -> Result<Self::Output>
Performs a forward pass through the model. Read more
Source§fn load_pretrained(&mut self, reader: &mut dyn Read) -> Result<()>
fn load_pretrained(&mut self, reader: &mut dyn Read) -> Result<()>
Loads pretrained weights into the model. Read more
Source§fn get_config(&self) -> &Self::Config
fn get_config(&self) -> &Self::Config
Returns a reference to the model’s configuration. Read more
Source§fn num_parameters(&self) -> usize
fn num_parameters(&self) -> usize
Returns the total number of parameters in the model. Read more
Auto Trait Implementations§
impl Freeze for RetNetForLanguageModeling
impl RefUnwindSafe for RetNetForLanguageModeling
impl Send for RetNetForLanguageModeling
impl Sync for RetNetForLanguageModeling
impl Unpin for RetNetForLanguageModeling
impl UnsafeUnpin for RetNetForLanguageModeling
impl UnwindSafe for RetNetForLanguageModeling
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more