pub struct RetNetForSequenceClassification { /* private fields */ }Expand description
RetNet for sequence classification
Implementations§
Source§impl RetNetForSequenceClassification
impl RetNetForSequenceClassification
pub fn new(config: RetNetConfig, num_labels: usize) -> Result<Self>
pub fn new_with_device( config: RetNetConfig, num_labels: usize, device: Device, ) -> Result<Self>
pub fn device(&self) -> Device
Trait Implementations§
Source§impl Model for RetNetForSequenceClassification
impl Model for RetNetForSequenceClassification
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 RetNetForSequenceClassification
impl RefUnwindSafe for RetNetForSequenceClassification
impl Send for RetNetForSequenceClassification
impl Sync for RetNetForSequenceClassification
impl Unpin for RetNetForSequenceClassification
impl UnsafeUnpin for RetNetForSequenceClassification
impl UnwindSafe for RetNetForSequenceClassification
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