pub struct GradientCheckpoint {
pub layer_index: usize,
pub activations: Vec<Box<dyn Any + Send + Sync>>,
pub timestamp: Instant,
pub memory_usage_mb: f64,
}Expand description
Gradient checkpoint for memory-efficient training
Fields§
§layer_index: usize§activations: Vec<Box<dyn Any + Send + Sync>>§timestamp: Instant§memory_usage_mb: f64Trait Implementations§
Auto Trait Implementations§
impl !RefUnwindSafe for GradientCheckpoint
impl !UnwindSafe for GradientCheckpoint
impl Freeze for GradientCheckpoint
impl Send for GradientCheckpoint
impl Sync for GradientCheckpoint
impl Unpin for GradientCheckpoint
impl UnsafeUnpin for GradientCheckpoint
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
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
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