pub struct MemoryReplay { /* private fields */ }Expand description
Memory replay optimizer.
Implementations§
Source§impl MemoryReplay
impl MemoryReplay
Sourcepub fn new(config: MemoryReplayConfig, initial_parameters: Vec<Tensor>) -> Self
pub fn new(config: MemoryReplayConfig, initial_parameters: Vec<Tensor>) -> Self
Create a new memory replay optimizer.
Sourcepub fn store_gradient(&mut self, gradients: &[Tensor]) -> Result<()>
pub fn store_gradient(&mut self, gradients: &[Tensor]) -> Result<()>
Add gradient to memory buffer.
Sourcepub fn step(&mut self, gradients: &[Tensor]) -> Result<()>
pub fn step(&mut self, gradients: &[Tensor]) -> Result<()>
Perform optimization step with memory replay.
Sourcepub fn get_parameters(&self) -> &[Tensor]
pub fn get_parameters(&self) -> &[Tensor]
Get current parameters.
Sourcepub fn memory_size(&self) -> usize
pub fn memory_size(&self) -> usize
Get memory buffer size.
Auto Trait Implementations§
impl Freeze for MemoryReplay
impl RefUnwindSafe for MemoryReplay
impl Send for MemoryReplay
impl Sync for MemoryReplay
impl Unpin for MemoryReplay
impl UnwindSafe for MemoryReplay
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