pub struct RetNetModel { /* private fields */ }Expand description
Main RetNet model
Implementations§
Source§impl RetNetModel
impl RetNetModel
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 RetNetModel
impl Model for RetNetModel
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 RetNetModel
impl RefUnwindSafe for RetNetModel
impl Send for RetNetModel
impl Sync for RetNetModel
impl Unpin for RetNetModel
impl UnsafeUnpin for RetNetModel
impl UnwindSafe for RetNetModel
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