pub struct CreativeWritingModel { /* private fields */ }Expand description
Creative writing model implementation
Implementations§
Source§impl CreativeWritingModel
impl CreativeWritingModel
pub fn new(config: CreativeWritingConfig) -> Result<Self>
Trait Implementations§
Source§impl Layer for CreativeWritingModel
impl Layer for CreativeWritingModel
Source§impl Model for CreativeWritingModel
impl Model for CreativeWritingModel
type Config = CreativeWritingConfig
type Input = Tensor
type Output = Tensor
Source§fn forward(&self, input: Self::Input) -> CoreResult<Self::Output>
fn forward(&self, input: Self::Input) -> CoreResult<Self::Output>
Performs a forward pass through the model. Read more
Source§fn load_pretrained(&mut self, reader: &mut dyn Read) -> CoreResult<()>
fn load_pretrained(&mut self, reader: &mut dyn Read) -> CoreResult<()>
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 CreativeWritingModel
impl RefUnwindSafe for CreativeWritingModel
impl Send for CreativeWritingModel
impl Sync for CreativeWritingModel
impl Unpin for CreativeWritingModel
impl UnsafeUnpin for CreativeWritingModel
impl UnwindSafe for CreativeWritingModel
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