pub enum CheckpointStrategy {
Uniform,
MemoryBased,
Custom,
SqrtStrategy,
}Expand description
Gradient checkpointing strategy.
Variants§
Uniform
Checkpoint every N layers uniformly.
MemoryBased
Checkpoint based on memory estimates.
Custom
Custom checkpointing (user-specified layers).
SqrtStrategy
Square root strategy (checkpoint sqrt(L) layers for L total).
Trait Implementations§
Source§impl Clone for CheckpointStrategy
impl Clone for CheckpointStrategy
Source§fn clone(&self) -> CheckpointStrategy
fn clone(&self) -> CheckpointStrategy
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreimpl Copy for CheckpointStrategy
Source§impl Debug for CheckpointStrategy
impl Debug for CheckpointStrategy
impl Eq for CheckpointStrategy
Source§impl PartialEq for CheckpointStrategy
impl PartialEq for CheckpointStrategy
Source§fn eq(&self, other: &CheckpointStrategy) -> bool
fn eq(&self, other: &CheckpointStrategy) -> bool
Tests for
self and other values to be equal, and is used by ==.impl StructuralPartialEq for CheckpointStrategy
Auto Trait Implementations§
impl Freeze for CheckpointStrategy
impl RefUnwindSafe for CheckpointStrategy
impl Send for CheckpointStrategy
impl Sync for CheckpointStrategy
impl Unpin for CheckpointStrategy
impl UnsafeUnpin for CheckpointStrategy
impl UnwindSafe for CheckpointStrategy
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> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§impl<Q, K> Equivalent<K> for Q
impl<Q, K> Equivalent<K> for Q
Source§fn equivalent(&self, key: &K) -> bool
fn equivalent(&self, key: &K) -> bool
Compare self to
key and return true if they are equal.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